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()
1300 QCHECK_EQ(block_shape_array_shape.dimensions_count(), 1); in ProcessSpaceToBatchNDOperator()
1301 QCHECK_EQ(paddings_array_shape.dimensions_count(), 2); in ProcessSpaceToBatchNDOperator()
1304 QCHECK_EQ(block_shape_array_shape.dims(0), spatial_dims_num); in ProcessSpaceToBatchNDOperator()
1312 QCHECK_EQ(paddings_array_shape.dims(0), spatial_dims_num); in ProcessSpaceToBatchNDOperator()
1313 QCHECK_EQ(paddings_array_shape.dims(1), 2); // Two parameters per dimension. in ProcessSpaceToBatchNDOperator()
1327 QCHECK_EQ(final_dim_size % block_shape_data[dim], 0); in ProcessSpaceToBatchNDOperator()
1354 QCHECK_EQ(block_shape_array_shape.dimensions_count(), 1); in ProcessBatchToSpaceNDOperator()
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_allocator.cc34 QCHECK_EQ(0, alignment & (alignment - 1)) << "Alignment must be power of 2."; in Align()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/
Dweighted_quantiles_stream.h215 QCHECK_EQ(buffer_.Size(), 0); in SerializeInternalSummaries()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc308 QCHECK_EQ(info.engine_name, (**node).name()) in UpdateToEngineNode()
359 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
546 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
Dconvert_nodes_test.cc1359 QCHECK_EQ(0, cudaStreamCreate(&stream_)); in OpConverterTest()
1363 ~OpConverterTest() override { QCHECK_EQ(0, cudaStreamDestroy(stream_)); } in ~OpConverterTest()
1587 QCHECK_EQ(num_elements, values.size()) in AddTestWeights()
1592 QCHECK_EQ(weights.size_bytes(), sizeof(T) * values.size()) in AddTestWeights()
4662 QCHECK_EQ(4, tf_input_sizes.size()); in TEST_F()
4671 QCHECK_EQ(2, tf_input_sizes.size()); in TEST_F()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc1549 QCHECK_EQ(dst_input_array->field_name(), \ in ResolveModelFlags()
1567 QCHECK_EQ(specified_input_array.shape().dims_size(), in ResolveModelFlags()
1579 QCHECK_EQ(specified_input_array.shape().dims(i), in ResolveModelFlags()
1607 QCHECK_EQ(model_flags.name(), model->flags.name()) \ in ResolveModelFlags()
2426 QCHECK_EQ(CountOpsWithInput(*model, weights_name), 1); in UndoWeightsShuffling()
2432 QCHECK_EQ(weights_shape.dimensions_count(), 2); in UndoWeightsShuffling()
2435 QCHECK_EQ(rows % 4, 0); in UndoWeightsShuffling()
2436 QCHECK_EQ(cols % 16, 0); in UndoWeightsShuffling()
Dtoco_cmdline_flags.cc324 QCHECK_EQ(input_types[i], input_types[0]); in ReadTocoFlagsFromCommandLineFlags()
Dimport_tensorflow.cc2097 QCHECK_EQ(num_inputs, 1); in ConvertUnpackOperator()
/external/tensorflow/tensorflow/core/platform/default/
Dlogging.h452 #define QCHECK_EQ(x, y) CHECK_EQ(x, y) macro