| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| H A D | benchmark_mlir_function.cc | 51 llvm::ArrayRef<InputTensorSpec> input_specs) { in GetInputTensors()
82 llvm::ArrayRef<InputTensorSpec> input_specs, in RunJitRtBenchmark()
171 ArrayRef<InputTensorSpec> input_specs) { in RunTfrtBenchmark()
195 llvm::ArrayRef<InputTensorSpec> input_specs) { in RunEigenBenchmark()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | python_api_parameter_converter_test.py | 63 input_specs, argument
439 def testConvert(self, param_names, input_specs, attr_specs, inputs, outputs, argument
468 input_specs, argument
|
| H A D | python_api_info_test.py | 49 input_specs, argument
243 def testInitializeFromParamSpecs(self, api_name, param_names, input_specs, argument
|
| H A D | python_api_info_wrapper.cc | 39 const std::map<std::string, std::string>& input_specs, in InitializeFromParamSpecs()
|
| H A D | python_api_info.cc | 208 const std::map<std::string, std::string>& input_specs, in InitializeFromParamSpecs()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/ |
| H A D | context.h | 147 const std::vector<InputSpec>& input_specs() const { in input_specs() function
151 void set_input_specs(const std::vector<InputSpec>& input_specs) { in set_input_specs()
|
| H A D | context.cpp | 171 std::vector<c10::IValue> input_specs; in serialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/ |
| H A D | rpc_ops.cc | 163 StructuredValue input_specs; member
184 const StructuredValue& input_specs, in Register()
324 const StructuredValue& input_specs, in Register()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
| H A D | saved_model.h | 53 input_specs; member
|
| H A D | saved_model.cc | 84 input_specs; in MapFunctionSignaturesFromTFSavedModelMLIR() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/proto/ |
| H A D | tf_rpc_service.proto | 21 StructuredValue input_specs = 2; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| H A D | device_target.cc | 110 llvm::SmallVector<Type, 4> input_specs, out_specs; in DecomposeMultiplyAccumulateScale() local
|
| H A D | quantization_context.cc | 128 llvm::SmallVector<Attribute, 4> input_specs; in Finalize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| H A D | saved_model.h | 72 std::vector<TensorSpec> input_specs; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | pmap_lib.cc | 173 std::vector<InputSpec> input_specs; member
478 std::vector<InputSpec>& input_specs = cache_entry.input_specs; in Call() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/experimental/rpc/ |
| H A D | rpc_ops.py | 367 def _add_method(self, method_name, output_specs, input_specs, client_handle, argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ |
| H A D | _op_schema.py | 83 input_specs: Optional[Sequence[DTensorSpec]] = None variable in PlacementStrategy
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| H A D | saved_model_utils.cc | 159 std::vector<const TensorSpecProto*> input_specs; in ValidateSavedFunctionCompatibleWithFunctionDef() local
|