Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
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
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
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
/external/tflite-support/tensorflow_lite_support/codegen/
Dcode_generator.cc74 std::vector<std::string> input_names; in NameInputsAndOutputs() local
149 auto& input_names = *inputs; in ResolveConflictedInputAndOutputNames() local
/external/tensorflow/tensorflow/tools/graph_transforms/
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
Dtransform_utils.h223 std::vector<string> input_names; member
/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/
Dsaved_model.h50 llvm::ArrayRef<llvm::StringRef> input_names; member
Dsaved_model.cc81 llvm::SmallVector<llvm::StringRef, 4> input_names; in MapFunctionSignaturesFromTFSavedModelMLIR() local
/external/tensorflow/tensorflow/lite/
Dsignature_runner.h100 const std::vector<const char*>& input_names() { return input_names_; } in input_names() function
Dsignature_runner_test.cc52 const std::vector<const char*>& input_names = add_runner->input_names(); in TEST() local
/external/armnn/python/pyarmnn/examples/image_classification/
Dtflite_mobilenetv1_quantized.py36 input_names = parser.GetSubgraphInputTensorNames(graph_id) variable
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model_testutil.cc95 const std::vector<std::string>& input_names, in ComputeCurrentTFResult()
138 const std::vector<std::string>& input_names, in ComputeCurrentTFResult()
Dsaved_model.h71 std::vector<std::string> input_names; member
/external/tensorflow/tensorflow/core/debug/
Ddebugger_state_impl.cc43 const int64_t executor_step_index, const std::vector<string>& input_names, in PublishDebugMetadata()
/external/tensorflow/tensorflow/lite/c/
Dc_api_signature_runner_test.cc54 std::vector<const char*> input_names; in TEST() local
/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py26 input_names=None, argument
Dmlir_wrapper.cc55 const std::string &output_names) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/common_runtime/
Dsession.cc46 Status Session::PRunSetup(const std::vector<string>& input_names, in PRunSetup()
Dgraph_runner.cc132 std::vector<string> input_names; in Run() local
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util_v2.py146 def resource_input_index(tensor_name, input_names, node_defs, functions): argument
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc82 auto input_names = input_attr.cast<mlir::StringAttr>().getValue(); in HandleInputOutputArraysWithModule() local
Djax_to_tfl_flatbuffer.cc177 std::vector<std::string> input_names; in ConvertJaxToTFLiteFlatBuffer() local
/external/tensorflow/tensorflow/dtensor/mlir/
Dhandle_sparsetensors.cc90 llvm::SmallVector<llvm::StringRef, 2> input_names; in UpdateFunctionInputAttributes() local
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dinsert_main_function.cc115 std::vector<std::string> input_names, output_names; in CreateMainFunction() local
/external/tensorflow/tensorflow/core/protobuf/
Ddebug_event.proto182 repeated string input_names = 6; field
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op_test.cc107 std::vector<string> input_names; in TEST_P() local

12