Searched defs:input_def (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | converter.cc | 120 absl::Status Init(const TensorObjectDef& input_def, in Init()
208 absl::Status Init(const TensorObjectDef& input_def, in Init()
291 const TensorObjectDef& input_def, in GetFromBhwcKernel()
304 absl::Status Init(const TensorObjectDef& input_def, in Init()
415 absl::Status Init(const TensorObjectDef& input_def, in Init()
479 absl::Status Init(const TensorObjectDef& input_def, in Init()
532 const auto& input_def = input.object_def; in IsSupported() local
546 const auto& input_def = input.object_def; in MakeConverter() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | converter.cc | 106 absl::Status Init(const TensorObjectDef& input_def, in Init()
194 absl::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
|
| D | converter_test.cc | 71 TensorObjectDef input_def; in RunFromTensorTest() local
128 TensorObjectDef input_def; in RunToTensorTest() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation.cc | 356 const auto& input_def = op_def_->input_arg(inference_arg_idx_++); in MaybeInferSingleInputAttrs() local
377 const OpDef::ArgDef& input_def, const DataType dtype, int num_inputs) { in InferSingleTypeInputListAttrs()
390 const OpDef::ArgDef& input_def, const std::vector<DataType>& dtypes) { in InferMixedTypeInputListAttrs()
404 const auto& input_def = op_def_->input_arg(inference_arg_idx_++); in InferInputListAttrs() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | replicate_invariant_op_hoisting.cc | 90 Operation* input_def = input.getDefiningOp(); in MakeShapeOpInvariant() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | gl_interop.cc | 263 GlClBufferCopier::GlClBufferCopier(const TensorObjectDef& input_def, in GlClBufferCopier()
|
| /external/tensorflow/tensorflow/java/src/gen/cc/ |
| D | op_specs.cc | 285 ArgumentSpec CreateInput(const OpDef_ArgDef& input_def, in CreateInput()
|