Home
last modified time | relevance | path

Searched refs:TensorInfo (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/go/
Dsignature_test.go30 Inputs: map[string]*corepb.TensorInfo{
31 "input_1": &corepb.TensorInfo{
44 "input_2": &corepb.TensorInfo{
58 Outputs: map[string]*corepb.TensorInfo{
59 "output_1": &corepb.TensorInfo{
72 "output_2": &corepb.TensorInfo{
90 Inputs: map[string]TensorInfo{
91 "input_1": TensorInfo{
96 "input_2": TensorInfo{
102 Outputs: map[string]TensorInfo{
[all …]
Dsignature.go82 Inputs, Outputs map[string]TensorInfo
87 type TensorInfo struct { struct
94 inputs := make(map[string]TensorInfo)
98 outputs := make(map[string]TensorInfo)
109 func tensorInfoFromProto(pb *corepb.TensorInfo) TensorInfo { argument
114 return TensorInfo{
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-tensor-info.pbtxt1 path: "tensorflow.TensorInfo"
4 name: "TensorInfo"
17 type_name: ".tensorflow.TensorInfo.CooSparse"
25 type_name: ".tensorflow.TensorInfo.CompositeTensor"
77 type_name: ".tensorflow.TensorInfo"
Dtensorflow.-tensor-info.-composite-tensor.pbtxt1 path: "tensorflow.TensorInfo.CompositeTensor"
17 type_name: ".tensorflow.TensorInfo"
Dtensorflow.-tensor-info.-coo-sparse.pbtxt1 path: "tensorflow.TensorInfo.CooSparse"
/external/tflite-support/tensorflow_lite_support/codegen/
Dandroid_java_generator.h36 struct TensorInfo { struct
58 std::vector<TensorInfo> inputs; argument
59 std::vector<TensorInfo> outputs;
Dandroid_java_generator.cc60 using details_android_java::TensorInfo;
107 TensorInfo CreateTensorInfo(const TensorMetadata* metadata, in CreateTensorInfo()
110 TensorInfo tensor_info; in CreateTensorInfo()
215 const TensorInfo& tensor_info) { in SetCodeWriterWithTensorInfo()
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_output_test.py85 meta_graph_pb2.TensorInfo(name='input-tensor-1:0',
90 meta_graph_pb2.TensorInfo(name='output-tensor-1:0',
117 meta_graph_pb2.TensorInfo(name='input-tensor-1:0',
122 meta_graph_pb2.TensorInfo(name='output-tensor-1:0',
154 meta_graph_pb2.TensorInfo(name='input-tensor-1:0',
159 meta_graph_pb2.TensorInfo(name='output-tensor-classes:0',
164 meta_graph_pb2.TensorInfo(name='output-tensor-scores:0',
195 meta_graph_pb2.TensorInfo(name='input-tensor-1:0',
200 meta_graph_pb2.TensorInfo(name='output-tensor-scores:0',
/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/
DInterpreter.cs42 public struct TensorInfo { struct in TensorFlowLite.Interpreter
124 public TensorInfo GetInputTensorInfo(int index) { in GetInputTensorInfo()
129 public TensorInfo GetOutputTensorInfo(int index) { in GetOutputTensorInfo()
147 private static TensorInfo GetTensorInfo(TfLiteTensor tensor) { in GetTensorInfo()
152 return new TensorInfo() { in GetTensorInfo()
/external/tensorflow/tensorflow/core/protobuf/
Dmeta_graph.proto32 // TensorInfo
217 message TensorInfo { message
222 // the SparseTensor as a whole, given in the enclosing TensorInfo.
238 // A TensorInfo for each flattened component tensor.
239 repeated TensorInfo components = 2;
319 map<string, TensorInfo> inputs = 1;
321 map<string, TensorInfo> outputs = 2;
337 TensorInfo tensor_info = 1;
/external/tensorflow/tensorflow/lite/tools/signature/
Dsignature_def_utils_test.py44 sig_input_tensor = meta_graph_pb2.TensorInfo(
48 sig_output_tensor = meta_graph_pb2.TensorInfo(
Dsignature_def_util_test.cc51 tensorflow::TensorInfo input_tensor; in GetTestSignatureDef()
52 tensorflow::TensorInfo output_tensor; in GetTestSignatureDef()
/external/tensorflow/tensorflow/python/saved_model/
Dutils_impl.py76 tensor_info = meta_graph_pb2.TensorInfo(
91 tensor_info = meta_graph_pb2.TensorInfo()
136 return meta_graph_pb2.TensorInfo(
Dsignature_def_utils_test.py36 _STRING = meta_graph_pb2.TensorInfo(
42 _FLOAT = meta_graph_pb2.TensorInfo(
Dsaved_model_test.py626 tensor_without_encoding = meta_graph_pb2.TensorInfo()
631 tensor_without_dtype = meta_graph_pb2.TensorInfo()
636 tensor_empty = meta_graph_pb2.TensorInfo()
640 valid_tensor_info = meta_graph_pb2.TensorInfo()
650 tensor_with_name = meta_graph_pb2.TensorInfo()
665 tensor_with_coo = meta_graph_pb2.TensorInfo()
/external/tensorflow/tensorflow/python/
Dmodules_with_exports.py33 from tensorflow.core.protobuf.meta_graph_pb2 import TensorInfo
95 tf_export(v1=['TensorInfo'])(TensorInfo)
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder_test.cc221 TensorInfo input, output; in TEST_F()
231 TensorInfo input2, output2; in TEST_F()
262 TensorInfo input, output; in TEST_F()
288 TensorInfo input, output; in TEST_F()
Dgrappler_item_builder.cc326 const TensorInfo& input = name_and_input.second; in GrapplerItemFromMetaGraphDef()
381 const TensorInfo& output = name_and_output.second; in GrapplerItemFromMetaGraphDef()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dno_input_shape_v1.py52 tensor_info_x = meta_graph_pb2.TensorInfo(
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc434 TensorInfo& in = (*signature_def.mutable_inputs())["input_arg"]; in TEST_F()
438 TensorInfo& out = (*signature_def.mutable_outputs())["output_arg"]; in TEST_F()
461 TensorInfo& in = (*signature_def.mutable_inputs())["input_arg"]; in TEST_F()
466 TensorInfo& out = (*signature_def.mutable_outputs())["output_arg"]; in TEST_F()
Dfreeze_saved_model.cc36 void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info, in GetTensorNamesFromTensorInfo()
/external/tensorflow/tensorflow/core/kernels/
Ddecode_proto_op.cc883 struct TensorInfo { in AccumulateFields() struct
884 explicit TensorInfo(Tensor* tensor) { in AccumulateFields() argument
914 std::vector<TensorInfo> tensors; in AccumulateFields()
926 const TensorInfo& info = tensors[output_index]; in AccumulateFields()
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py89 def _generate_random_tensor_v1(tensor_info: meta_graph_pb2.TensorInfo, argument
272 def input_tensor_info(self) -> Mapping[str, meta_graph_pb2.TensorInfo]:
277 def output_tensor_info(self) -> Mapping[str, meta_graph_pb2.TensorInfo]:
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc3500 const std::vector<std::pair<std::string, TensorInfo>>& inputs,
3501 const std::vector<std::pair<std::string, TensorInfo>>& outputs,
3510 llvm::ArrayRef<std::pair<std::string, TensorInfo>> inputs);
3579 std::vector<std::pair<std::string, TensorInfo>> inputs; in ConvertInitializer()
3582 TensorInfo tensor_info; in ConvertInitializer()
3621 const std::vector<std::pair<std::string, TensorInfo>>& inputs, in ConvertGraph()
3622 const std::vector<std::pair<std::string, TensorInfo>>& outputs, in ConvertGraph()
3646 std::vector<std::pair<std::string, TensorInfo>> inputs( in ConvertSignature()
3651 std::vector<std::pair<std::string, TensorInfo>> outputs( in ConvertSignature()
3690 llvm::ArrayRef<std::pair<std::string, TensorInfo>> inputs) { in ParseInputArrays()
[all …]
/external/XNNPACK/bench/
Dconvolution.cc865 arm_compute::TensorInfo input_info( in armcl_convolution_f32()
880 arm_compute::TensorInfo kernel_info( in armcl_convolution_f32()
890 arm_compute::TensorInfo bias_info( in armcl_convolution_f32()
905 arm_compute::TensorInfo output_info( in armcl_convolution_f32()

12