Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc49 const std::vector<std::string>& output_arrays, in GraphdefToMlirImport()
107 const std::vector<std::string>& output_arrays, in GraphdefToMlirTranslateFunction()
126 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToMlirTranslateFunction()
221 const std::vector<std::string>& output_arrays, in GraphdefToSplattedMlirTranslateFunction()
270 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate_cl.cc46 opt<std::string> output_arrays( variable
/external/tensorflow/tensorflow/lite/python/
Dlite.py1505 output_arrays=None, argument
1633 output_arrays=None, argument
1793 output_arrays=None, argument
1845 output_arrays, argument
1939 output_arrays=None, argument
1987 output_arrays=None, argument
2046 output_arrays, argument
2059 output_arrays=None, argument
2074 output_arrays=None): argument
Dconvert_saved_model.py156 output_arrays, tag_set, signature_key): argument
Dconvert_saved_model_test.py86 output_arrays=None, argument
Dutil.py252 output_arrays, argument
Dconvert.py541 def toco_convert_graph_def(input_data, input_arrays_with_shape, output_arrays, argument
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py327 def evaluate_frozen_graph(filename, input_arrays, output_arrays): argument
589 output_arrays, argument
671 output_arrays, argument
880 output_arrays, argument
/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py33 output_arrays, argument
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc71 std::vector<string> output_arrays(model_flags.output_arrays().begin(), in ConvertGraphDefToTFLiteFlatBuffer() local
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc77 absl::string_view output_arrays, bool prune_unused_nodes, in LoadFromGraphdefOrMlirSource()
/external/tensorflow/tensorflow/lite/toco/
Dmodel_flags.proto148 repeated string output_arrays = 2; field
Dmodel_cmdline_flags.cc220 std::vector<std::string> output_arrays = in ReadModelFlagsFromCommandLineFlags() local
Dargs.h129 Arg<std::string> output_arrays; member
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc102 absl::string_view input_shapes, absl::string_view output_arrays, in LoadFromGraphdefOrMlirSource()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc672 std::vector<llvm_ir::IrArray> output_arrays; in ConstructIrArrayForOutputs() local
Dir_emitter_unnested.cc4146 absl::Span<const llvm_ir::IrArray> output_arrays, in EmitTileElementForFusion()
4819 absl::Span<const llvm_ir::IrArray> output_arrays, in EmitHlo021Tile()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc3144 std::vector<llvm_ir::IrArray> output_arrays; in EmitTargetElementLoop() local