MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / input

Method input

tensorflow/core/framework/op_kernel.cc:351–367  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

349}
350
351Status OpKernelContext::input(StringPiece name, const Tensor** tensor) {
352 int start, stop;
353 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop));
354 if (stop != start + 1) {
355 return errors::InvalidArgument("OpKernel used list-valued input name '",
356 name,
357 "' when single-valued input was "
358 "expected");
359 }
360 if (input_is_ref(start)) {
361 return errors::InvalidArgument("OpKernel used ref input name '", name,
362 "' when non-ref input was expected");
363 }
364 *tensor = (*params_->inputs)[start].tensor;
365 record_tensor_reference(**tensor);
366 return Status::OK();
367}
368
369Status OpKernelContext::input_dtype(StringPiece name, DataType* dtype) const {
370 int start, stop;

Callers 15

SummarizeNodeFunction · 0.45
ComputeAsyncMethod · 0.45
mpi_ops.ccFile · 0.45
ComputeAsyncMethod · 0.45
mpi_ops.ccFile · 0.45
ComputeAsyncMethod · 0.45
ComputeAsyncMethod · 0.45
array_ops.ccFile · 0.45
ComputeMethod · 0.45
ComputeMethod · 0.45
variable_ops.ccFile · 0.45
mainFunction · 0.45

Calls 2

InvalidArgumentFunction · 0.85
InputRangeMethod · 0.80

Tested by 15

CheckRegressionSignatureFunction · 0.36
TESTFunction · 0.36
EqualFunctionNodeDefFunction · 0.36
EqualFunctionNodeDefFunction · 0.36
TEST_FFunction · 0.36
RFFTShapeFunction · 0.36
TESTFunction · 0.36
InferShapesMethod · 0.36
TESTFunction · 0.36
ComputeMethod · 0.36
TEST_FFunction · 0.36