Home
last modified time | relevance | path

Searched defs:input_def (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc60 const auto& input_def = op_def_->input_arg(inference_arg_idx_++); in MaybeInferSingleInputAttrs() local
81 const tensorflow::OpDef::ArgDef& input_def, in InferSingleTypeInputListAttrs()
95 const tensorflow::OpDef::ArgDef& input_def, in InferMixedTypeInputListAttrs()
110 const auto& input_def = op_def_->input_arg(inference_arg_idx_++); in InferInputListAttrs() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc88 const TensorObjectDef& input_def, in GetToDhwc4Kernel()
102 const TensorObjectDef& input_def, in GetToBhwcKernel()
122 Status Init(const TensorObjectDef& input_def, in Init()
203 const TensorObjectDef& input_def, in GetFromDhwc4Kernel()
215 const TensorObjectDef& input_def, in GetFromBhwcKernel()
228 Status Init(const TensorObjectDef& input_def, in Init()
321 Status Init(const TensorObjectDef& input_def, in Init()
383 Status Init(const TensorObjectDef& input_def, in Init()
436 const auto& input_def = input.object_def; in IsSupported() local
449 const auto& input_def = input.object_def; in MakeConverter() local
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconverter.cc106 Status Init(const TensorObjectDef& input_def, in Init()
194 Status Init(const TensorObjectDef& input_def, in Init()
349 const auto& input_def = input.object_def; in IsSupported() local
362 const auto& input_def = input.object_def; in MakeConverter() local
Dconverter_test.cc71 TensorObjectDef input_def; in RunFromTensorTest() local
128 TensorObjectDef input_def; in RunToTensorTest() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dreplicate_invariant_op_hoisting.cc86 Operation* input_def = input.getDefiningOp(); in MakeShapeOpInvariant() local
/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_specs.cc285 ArgumentSpec CreateInput(const OpDef_ArgDef& input_def, in CreateInput()
/external/v8/src/torque/
Dcc-generator.cc32 DefinitionLocation input_def = block->InputDefinitions().Peek(i); in EmitGraph() local