Home
last modified time | relevance | path

Searched refs:QCHECK_EQ (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tools/
Dconvert_computation.cc57 QCHECK_EQ(argc, 3) << "usage: " << argv[0] << " <txt2bin|bin2txt> <path>"; in main()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc312 QCHECK_EQ(depth % (block_size * block_size), 0); in ProcessDepthToSpaceOperator()
335 QCHECK_EQ(width % block_size, 0); in ProcessSpaceToDepthOperator()
336 QCHECK_EQ(height % block_size, 0); in ProcessSpaceToDepthOperator()
1301 QCHECK_EQ(block_shape_array_shape.dimensions_count(), 1); in ProcessSpaceToBatchNDOperator()
1302 QCHECK_EQ(paddings_array_shape.dimensions_count(), 2); in ProcessSpaceToBatchNDOperator()
1305 QCHECK_EQ(block_shape_array_shape.dims(0), 2); in ProcessSpaceToBatchNDOperator()
1315 QCHECK_EQ(paddings_array_shape.dims(0), 2); // Number of block dimensions in ProcessSpaceToBatchNDOperator()
1316 QCHECK_EQ(paddings_array_shape.dims(1), 2); // Two parameters per dimension. in ProcessSpaceToBatchNDOperator()
1325 QCHECK_EQ(height_with_paddings % block_height, 0); in ProcessSpaceToBatchNDOperator()
1326 QCHECK_EQ(width_with_paddings % block_width, 0); in ProcessSpaceToBatchNDOperator()
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_allocator.cc36 QCHECK_EQ(0, alignment & (alignment - 1)) << "Alignment must be power of 2."; in Align()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc1546 QCHECK_EQ(dst_input_array->field_name(), \ in ResolveModelFlags()
1564 QCHECK_EQ(specified_input_array.shape().dims_size(), in ResolveModelFlags()
1576 QCHECK_EQ(specified_input_array.shape().dims(i), in ResolveModelFlags()
1604 QCHECK_EQ(model_flags.name(), model->flags.name()) \ in ResolveModelFlags()
2406 QCHECK_EQ(CountOpsWithInput(*model, weights_name), 1); in UndoWeightsShuffling()
2412 QCHECK_EQ(weights_shape.dimensions_count(), 2); in UndoWeightsShuffling()
2415 QCHECK_EQ(rows % 4, 0); in UndoWeightsShuffling()
2416 QCHECK_EQ(cols % 16, 0); in UndoWeightsShuffling()
Dtoco_cmdline_flags.cc324 QCHECK_EQ(input_types[i], input_types[0]); in ReadTocoFlagsFromCommandLineFlags()
Dimport_tensorflow.cc2052 QCHECK_EQ(num_inputs, 1); in ConvertUnpackOperator()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc302 QCHECK_EQ(info.engine_name, (**node).name()) in UpdateToEngineNode()
353 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
516 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
Dconvert_nodes_test.cc1247 QCHECK_EQ(0, cudaStreamCreate(&stream_)); in OpConverterTest()
1251 ~OpConverterTest() override { QCHECK_EQ(0, cudaStreamDestroy(stream_)); } in ~OpConverterTest()
1396 QCHECK_EQ(num_elements, values.size()) in AddTestWeights()
1401 QCHECK_EQ(weights.size_bytes(), sizeof(T) * values.size()) in AddTestWeights()
3975 QCHECK_EQ(4, tf_input_sizes.size()); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/
Dweighted_quantiles_stream.h215 QCHECK_EQ(buffer_.Size(), 0); in SerializeInternalSummaries()
/external/tensorflow/tensorflow/core/platform/default/
Dlogging.h455 #define QCHECK_EQ(x, y) CHECK_EQ(x, y) macro