Home
last modified time | relevance | path

Searched defs:input_arrays (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_concatenation.cc37 void CopyTensorSegments(const std::vector<Array*>& input_arrays, in CopyTensorSegments()
80 void ConcatenateTensorBuffers(const std::vector<Array*>& input_arrays, in ConcatenateTensorBuffers()
111 const std::vector<Array*>& input_arrays, in SetMinMaxForConcatenedArray()
172 std::vector<Array*> input_arrays; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dtf_mlir_translate.cc46 const std::vector<std::string>& input_arrays, in GraphdefToMlirImport()
106 const std::vector<std::string>& input_arrays, in GraphdefToMlirTranslateFunction()
127 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirTranslateFunction()
244 const std::vector<std::string>& input_arrays, in GraphdefToSplattedMlirTranslateFunction()
296 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToSplattedMlirTranslateFunction()
H A Dtf_mlir_translate_cl.cc26 opt<std::string> input_arrays( variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dlite.py2241 input_arrays=None, argument
2618 input_arrays, argument
2712 input_arrays=None, argument
2764 input_arrays=None, argument
2828 input_arrays, argument
2840 input_arrays=None, argument
2855 input_arrays=None, argument
H A Dconvert_saved_model.py155 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
H A Dconvert_saved_model_test.py80 input_arrays=None, argument
H A Dutil.py221 input_arrays, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
H A Dtf_to_tfjs_json.cc76 absl::string_view debug_info_file, absl::string_view input_arrays, in LoadFromGraphdefOrMlirSource()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dmodel_flags.proto144 repeated InputArray input_arrays = 1; field
H A Dargs.h129 Arg<std::string> input_arrays; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/logging/
H A Dconversion_log_util_test.cc144 InputArray input_arrays[3]; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dtf_to_tfl_flatbuffer.cc111 absl::string_view input_arrays, absl::string_view input_dtypes, in LoadFromGraphdefOrMlirSource()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3900 absl::Span<HloInstruction* const> input_arrays = reduce_window->inputs(); in HandleReduceWindow() local