Searched defs:input_def (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| H A D | converter.cc | 125 absl::Status Init(const TensorObjectDef& input_def, in Init()
221 absl::Status Init(const TensorObjectDef& input_def, in Init()
314 const TensorObjectDef& input_def, in GetFromBhwcKernel()
327 absl::Status Init(const TensorObjectDef& input_def, in Init()
448 absl::Status Init(const TensorObjectDef& input_def, in Init()
513 absl::Status Init(const TensorObjectDef& input_def, in Init()
625 const auto& input_def = input.object_def; in IsSupported() local
639 const auto& input_def = input.object_def; in MakeConverter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| H A D | converter.cc | 111 absl::Status Init(const TensorObjectDef& input_def, in Init()
199 absl::Status Init(const TensorObjectDef& input_def, in Init()
354 const auto& input_def = input.object_def; in IsSupported() local
367 const auto& input_def = input.object_def; in MakeConverter() local
|
| H A D | converter_test.cc | 71 TensorObjectDef input_def; in RunFromTensorTest() local
128 TensorObjectDef input_def; in RunToTensorTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | eager_operation.cc | 358 const auto& input_def = op_def_->input_arg(inference_arg_idx_++); in MaybeInferSingleInputAttrs() local
379 const OpDef::ArgDef& input_def, const DataType dtype, int num_inputs) { in InferSingleTypeInputListAttrs()
392 const OpDef::ArgDef& input_def, const std::vector<DataType>& dtypes) { in InferMixedTypeInputListAttrs()
406 const auto& input_def = op_def_->input_arg(inference_arg_idx_++); in InferInputListAttrs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | replicate_invariant_op_hoisting.cc | 60 Operation* input_def = input.getDefiningOp(); in MakeShapeOpInvariant() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| H A D | gl_interop.cc | 266 GlClBufferCopier::GlClBufferCopier(const TensorObjectDef& input_def, in GlClBufferCopier()
|
| /aosp_15_r20/external/tensorflow/tensorflow/java/src/gen/cc/ |
| H A D | op_specs.cc | 285 ArgumentSpec CreateInput(const OpDef_ArgDef& input_def, in CreateInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.cc | 1956 const auto& input_def = op_def_->input_arg(i); in MaybeInferInputAttrs() local
|