Home
last modified time | relevance | path

Searched defs:output_arrays (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dtf_mlir_translate.cc49 const std::vector<std::string>& output_arrays, in GraphdefToMlirImport()
109 const std::vector<std::string>& output_arrays, in GraphdefToMlirTranslateFunction()
128 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToMlirTranslateFunction()
247 const std::vector<std::string>& output_arrays, in GraphdefToSplattedMlirTranslateFunction()
297 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToSplattedMlirTranslateFunction()
H A Dtf_mlir_translate_cl.cc47 opt<std::string> output_arrays( variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dlite.py2243 output_arrays=None, argument
2382 output_arrays=None, argument
2563 output_arrays=None, argument
2619 output_arrays, argument
2714 output_arrays=None, argument
2766 output_arrays=None, argument
2829 output_arrays, argument
2842 output_arrays=None, argument
2857 output_arrays=None): argument
H A Dconvert_saved_model.py156 output_arrays, tag_set, signature_key): argument
H A Dconvert_saved_model_test.py82 output_arrays=None, argument
H A Dutil.py222 output_arrays, argument
H A Dconvert.py655 output_arrays, control_output_arrays, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dgraphdef_to_tfl_flatbuffer.cc74 std::vector<string> output_arrays(model_flags.output_arrays().begin(), in ConvertGraphDefToTFLiteFlatBuffer() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
H A Dtf_to_tfjs_json.cc78 absl::string_view output_arrays, bool prune_unused_nodes, in LoadFromGraphdefOrMlirSource()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dmodel_flags.proto148 repeated string output_arrays = 2; field
H A Dmodel_cmdline_flags.cc220 std::vector<std::string> output_arrays = in ReadModelFlagsFromCommandLineFlags() local
H A Dargs.h131 Arg<std::string> output_arrays; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dtf_to_tfl_flatbuffer.cc112 absl::string_view input_shapes, absl::string_view output_arrays, in LoadFromGraphdefOrMlirSource()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc1550 absl::Span<llvm_ir::IrArray> output_arrays) { in EnableLogicalIndexGenerationForOutput()
3516 absl::Span<const llvm_ir::IrArray> output_arrays, in EmitTileElementForTranspose()
3724 const IrEmitterUnnested::ReductionOutputMap& output_arrays, in GetOutputAddressForReduction()
3837 const ReductionOutputMap& output_arrays, in EmitReductionOutputForRowReduction()
3964 const ReductionOutputMap& output_arrays, in EmitReductionOutputForColumnReduction()
4204 absl::Span<const llvm_ir::IrArray> output_arrays, in EmitHlo021Tile()
H A Dir_emitter.cc662 std::vector<llvm_ir::IrArray> output_arrays; in ConstructIrArrayForOutputs() local
/aosp_15_r20/external/virglrenderer/src/
H A Dvrend_shader.h125 struct vrend_shader_io_array_info output_arrays; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc3347 std::vector<llvm_ir::IrArray> output_arrays; in EmitTargetElementLoop() local