/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
H A D | trt_convert_api.cc | 65 const std::vector<std::string>& input_names, in RunGrappler() 156 const std::vector<std::string>& input_names, in RunTfTrt() 194 Status RunSession(Session* session, const std::vector<std::string>& input_names, in RunSession() 227 const std::vector<std::string>& input_names, in Build() 390 const GraphDef& frozen_graph_def, const std::vector<string>& input_names, in ConvertAndBuild() 499 std::vector<std::string> input_names = GetNodeNames(signature.inputs()); in ConvertAndBuild() local
|
/aosp_15_r20/external/pytorch/torch/onnx/ |
H A D | utils.py | 567 input_names=None, argument 1079 input_names=None, argument 1215 input_names=None, argument 1467 input_names=None, argument 1668 def _set_input_and_output_names(graph, input_names, output_names): argument 1943 def _validate_dynamic_axes(dynamic_axes, model, input_names, output_names): argument
|
H A D | _experimental.py | 19 input_names: Optional[Sequence[str]] = None variable in ExportOptions
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
H A D | fold_constants_lib.cc | 121 std::unordered_set<string> input_names; in RewriteInputsAsPlaceholders() local 152 std::unordered_set<TensorId, TensorId::Hasher> input_names; in RemoveUnusedNodes() local 242 std::vector<TensorId> input_names; in FoldConstants() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | save_op.cc | 56 static const char* input_names[3] = {"basename", "shard", "num_shards"}; in Compute() local 79 static const char* input_names[2] = {"basename", "num_shards"}; in Compute() local
|
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/codegen/ |
H A D | code_generator.cc | 74 std::vector<std::string> input_names; in NameInputsAndOutputs() local 149 auto& input_names = *inputs; in ResolveConflictedInputAndOutputNames() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/ |
H A D | saved_model_testutil.cc | 95 const std::vector<std::string>& input_names, in ComputeCurrentTFResult() 138 const std::vector<std::string>& input_names, in ComputeCurrentTFResult()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
H A D | session_ref.cc | 256 const std::vector<string>& input_names, in RecordPRunSetup() 450 Status SessionRef::PRunSetup(const std::vector<string>& input_names, in PRunSetup()
|
H A D | tf_session_helper.cc | 73 NameVector input_names; in TF_Run_wrapper_helper() local 316 const NameVector& input_names, in TF_PRunSetup_wrapper()
|
/aosp_15_r20/external/executorch/backends/apple/coreml/runtime/delegate/ |
H A D | model_metadata.h | 45 std::vector<std::string> input_names; member
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
H A D | saved_model.h | 50 llvm::ArrayRef<llvm::StringRef> input_names; member
|
H A D | saved_model.cc | 81 llvm::SmallVector<llvm::StringRef, 4> input_names; in MapFunctionSignaturesFromTFSavedModelMLIR() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/ |
H A D | signature_runner.h | 100 const std::vector<const char*>& input_names() { return input_names_; } in input_names() function
|
H A D | signature_runner_test.cc | 52 const std::vector<const char*>& input_names = add_runner->input_names(); in TEST() local
|
/aosp_15_r20/external/armnn/python/pyarmnn/examples/image_classification/ |
H A D | tflite_mobilenetv1_quantized.py | 36 input_names = parser.GetSubgraphInputTensorNames(graph_id) variable
|
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
H A D | debugger_state_impl.cc | 43 const int64_t executor_step_index, const std::vector<string>& input_names, in PublishDebugMetadata()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
H A D | c_api_signature_runner_test.cc | 54 std::vector<const char*> input_names; in TEST() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ |
H A D | pywrap_mlir.py | 26 input_names=None, argument
|
H A D | mlir_wrapper.cc | 55 const std::string &output_names) { in PYBIND11_MODULE()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | session.cc | 46 Status Session::PRunSetup(const std::vector<string>& input_names, in PRunSetup()
|
H A D | graph_runner.cc | 132 std::vector<string> input_names; in Run() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | control_flow_util_v2.py | 146 def resource_input_index(tensor_name, input_names, node_defs, functions): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
H A D | saved_model_to_tfl_flatbuffer.cc | 82 auto input_names = input_attr.cast<mlir::StringAttr>().getValue(); in HandleInputOutputArraysWithModule() local
|
H A D | jax_to_tfl_flatbuffer.cc | 177 std::vector<std::string> input_names; in ConvertJaxToTFLiteFlatBuffer() local
|
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
H A D | handle_sparsetensors.cc | 90 llvm::SmallVector<llvm::StringRef, 2> input_names; in UpdateFunctionInputAttributes() local
|