Home
last modified time | relevance | path

Searched refs:data_types (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/
Dhvx_ops_support_checker_main.cc64 std::vector<DataType> data_types; in SummarizeNode() local
67 node_def, &data_types, &shapes); in SummarizeNode()
68 if (data_types.empty() || shapes.empty()) { in SummarizeNode()
71 CHECK_EQ(data_types.size(), shapes.size()); in SummarizeNode()
72 for (int i = 0; i < data_types.size(); ++i) { in SummarizeNode()
73 LOG(INFO) << " Output(" << i << "): " << DataType_Name(data_types.at(i)) in SummarizeNode()
/external/vixl/tools/test_generator/
Dparser.py34 from test_generator import data_types
70 return data_types.Operand(name, type_name, variants, default)
83 input_constructor = getattr(data_types, type_name)
230 operands.append(data_types.OperandWrapper(json_operand["name"],
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc175 std::vector<DataType> data_types; in LoadGraphFromProto() local
178 node->attrs(), &data_types, &shapes); in LoadGraphFromProto()
180 CHECK(data_types.size() > port); in LoadGraphFromProto()
181 graph_output_node_info.set_dtype(data_types.at(port)); in LoadGraphFromProto()
329 std::vector<DataType> data_types; in TransformGraphToAddAggregatedInputNode() local
337 data_types.emplace_back(input_node_info_list.at(i).second.dtype()); in TransformGraphToAddAggregatedInputNode()
348 data_types) in TransformGraphToAddAggregatedInputNode()
380 node_name, AGGREGATED_INPUT_NODE_NAME, i, data_types.at(i), graph, in TransformGraphToAddAggregatedInputNode()
383 std::vector<DataType> data_types; in TransformGraphToAddAggregatedInputNode() local
386 original_input_node->attrs(), &data_types, &shapes); in TransformGraphToAddAggregatedInputNode()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc367 const std::vector<DataType>& data_types, in AddOutputTensorShapeType() argument
369 AddNodeAttr(ATTR_OUTPUT_DATA_TYPES, data_types, node_def); in AddOutputTensorShapeType()
384 std::vector<DataType> data_types; in AddOutputTensorShapeTypeByTensorShapeMap() local
390 data_types.emplace(data_types.begin(), tst->first); in AddOutputTensorShapeTypeByTensorShapeMap()
396 AddOutputTensorShapeType(data_types, shapes, node_def); in AddOutputTensorShapeTypeByTensorShapeMap()
401 AttrSlice attrs, std::vector<DataType>* data_types, in GetOutputTensorShapeType() argument
404 if (data_types != nullptr) { in GetOutputTensorShapeType()
405 status = GetNodeAttr(attrs, ATTR_OUTPUT_DATA_TYPES, data_types); in GetOutputTensorShapeType()
412 if (status.ok() && data_types != nullptr) { in GetOutputTensorShapeType()
413 CHECK_EQ(data_types->size(), shapes->size()); in GetOutputTensorShapeType()
[all …]
Dremote_fused_graph_execute_utils_test.cc381 std::vector<DataType> data_types; in TEST() local
384 &data_types)); in TEST()
385 ASSERT_EQ(1, data_types.size()); in TEST()
386 EXPECT_EQ(DT_FLOAT, data_types.at(0)); in TEST()
400 std::vector<DataType> data_types; in TEST() local
403 &data_types)); in TEST()
404 ASSERT_EQ(1, data_types.size()); in TEST()
405 EXPECT_EQ(DT_FLOAT, data_types.at(0)); in TEST()
Dremote_fused_graph_execute_utils.h134 static bool AddOutputTensorShapeType(const std::vector<DataType>& data_types,
142 std::vector<DataType>* data_types,
/external/tensorflow/tensorflow/python/ops/
Dconv2d_benchmark.py176 data_types = [dtypes.float32, dtypes.float16]
185 data_types, data_formats, in_channels, out_channels, hw_strides,
/external/tensorflow/tensorflow/python/autograph/docs/
Dpyfunc_dtypes.md4 [data type](https://www.tensorflow.org/guide/tensors#data_types).
/external/vixl/test/aarch32/config/
Dcond-rd-memop-rs-shift-amount-1to32-a32.json125 // `rn`. See `data_types.MemOperand.Epilogue()` for details.
Dcond-rd-memop-immediate-8192-a32.json121 // `rn`. See `data_types.MemOperand.Epilogue()` for details.
Dcond-rd-memop-rs-shift-amount-1to31-a32.json133 // `rn`. See `data_types.MemOperand.Epilogue()` for details.
Dcond-rd-memop-immediate-512-a32.json121 // `rn`. See `data_types.MemOperand.Epilogue()` for details.
Dcond-rd-memop-rs-a32.json137 // `rn`. See `data_types.MemOperand.Epilogue()` for details.
Ddata-types.json1676 // of `test_generator.data_types.Inputs`. There should be a Python
/external/libxcam/xcore/
DMakefile.am133 interface/data_types.h \
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py126 def toco_options(data_types, argument
146 if data_types[0] == "QUANTIZED_UINT8":
148 s = (" --input_data_types=%s" % ",".join(data_types) +
344 data_types = [_TF_TYPE_INFO[x[2]][1] for x in input_tensors]
346 data_types=data_types,