Home
last modified time | relevance | path

Searched defs:input_names (Results 1 – 25 of 57) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.cc65 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 Dutils.py567 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.py19 input_names: Optional[Sequence[str]] = None variable in ExportOptions
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_constants_lib.cc121 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 Dsave_op.cc56 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 Dcode_generator.cc74 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 Dsaved_model_testutil.cc95 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 Dsession_ref.cc256 const std::vector<string>& input_names, in RecordPRunSetup()
450 Status SessionRef::PRunSetup(const std::vector<string>& input_names, in PRunSetup()
H A Dtf_session_helper.cc73 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 Dmodel_metadata.h45 std::vector<std::string> input_names; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/
H A Dsaved_model.h50 llvm::ArrayRef<llvm::StringRef> input_names; member
H A Dsaved_model.cc81 llvm::SmallVector<llvm::StringRef, 4> input_names; in MapFunctionSignaturesFromTFSavedModelMLIR() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dsignature_runner.h100 const std::vector<const char*>& input_names() { return input_names_; } in input_names() function
H A Dsignature_runner_test.cc52 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 Dtflite_mobilenetv1_quantized.py36 input_names = parser.GetSubgraphInputTensorNames(graph_id) variable
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebugger_state_impl.cc43 const int64_t executor_step_index, const std::vector<string>& input_names, in PublishDebugMetadata()
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dc_api_signature_runner_test.cc54 std::vector<const char*> input_names; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dpywrap_mlir.py26 input_names=None, argument
H A Dmlir_wrapper.cc55 const std::string &output_names) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dsession.cc46 Status Session::PRunSetup(const std::vector<string>& input_names, in PRunSetup()
H A Dgraph_runner.cc132 std::vector<string> input_names; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcontrol_flow_util_v2.py146 def resource_input_index(tensor_name, input_names, node_defs, functions): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dsaved_model_to_tfl_flatbuffer.cc82 auto input_names = input_attr.cast<mlir::StringAttr>().getValue(); in HandleInputOutputArraysWithModule() local
H A Djax_to_tfl_flatbuffer.cc177 std::vector<std::string> input_names; in ConvertJaxToTFLiteFlatBuffer() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dhandle_sparsetensors.cc90 llvm::SmallVector<llvm::StringRef, 2> input_names; in UpdateFunctionInputAttributes() local

123