Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dsparse_column_iterable.cc95 QCHECK_EQ(iter_, other.iter_); in operator !=()
104 QCHECK_EQ(iter_, other.iter_); in operator ==()
Dsparse_column_iterable.h63 QCHECK_EQ(iter_, other.iter_);
68 QCHECK_EQ(iter_, other.iter_);
Dexamples_iterable.h67 QCHECK_EQ(iter_, other.iter_);
72 QCHECK_EQ(iter_, other.iter_);
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc311 QCHECK_EQ(depth % (block_size * block_size), 0); in ProcessDepthToSpaceOperator()
334 QCHECK_EQ(width % block_size, 0); in ProcessSpaceToDepthOperator()
335 QCHECK_EQ(height % block_size, 0); in ProcessSpaceToDepthOperator()
1288 QCHECK_EQ(block_shape_array_shape.dimensions_count(), 1); in ProcessSpaceToBatchNDOperator()
1289 QCHECK_EQ(paddings_array_shape.dimensions_count(), 2); in ProcessSpaceToBatchNDOperator()
1292 QCHECK_EQ(block_shape_array_shape.dims(0), 2); in ProcessSpaceToBatchNDOperator()
1302 QCHECK_EQ(paddings_array_shape.dims(0), 2); // Number of block dimensions in ProcessSpaceToBatchNDOperator()
1303 QCHECK_EQ(paddings_array_shape.dims(1), 2); // Two parameters per dimension. in ProcessSpaceToBatchNDOperator()
1312 QCHECK_EQ(height_with_paddings % block_height, 0); in ProcessSpaceToBatchNDOperator()
1313 QCHECK_EQ(width_with_paddings % block_width, 0); in ProcessSpaceToBatchNDOperator()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Dconvert_computation.cc57 QCHECK_EQ(argc, 3) << "usage: " << argv[0] << " <txt2bin|bin2txt> <path>"; in main()
/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/contrib/boosted_trees/lib/models/
Dmultiple_additive_trees.cc65 QCHECK_EQ(leaf.index_size(), leaf.value_size()); in Predict()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc1530 QCHECK_EQ(dst_input_array->field_name(), \ in ResolveModelFlags()
1548 QCHECK_EQ(specified_input_array.shape().dims_size(), in ResolveModelFlags()
1560 QCHECK_EQ(specified_input_array.shape().dims(i), in ResolveModelFlags()
1588 QCHECK_EQ(model_flags.name(), model->flags.name()) \ in ResolveModelFlags()
2364 QCHECK_EQ(CountOpsWithInput(*model, weights_name), 1); in UndoWeightsShuffling()
2370 QCHECK_EQ(weights_shape.dimensions_count(), 2); in UndoWeightsShuffling()
2373 QCHECK_EQ(rows % 4, 0); in UndoWeightsShuffling()
2374 QCHECK_EQ(cols % 16, 0); in UndoWeightsShuffling()
Dtoco_cmdline_flags.cc307 QCHECK_EQ(input_types[i], input_types[0]); in ReadTocoFlagsFromCommandLineFlags()
Dimport_tensorflow.cc1972 QCHECK_EQ(num_inputs, 1); in ConvertUnpackOperator()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/
Dweighted_quantiles_stream.h213 QCHECK_EQ(buffer_.Size(), 0); in SerializeInternalSummaries()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/
Dweighted_quantiles_stream.h213 QCHECK_EQ(buffer_.Size(), 0); in SerializeInternalSummaries()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc377 QCHECK_EQ(info.engine_name, (**node).name()) in UpdateToEngineNode()
430 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
617 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
Dconvert_nodes_test.cc1070 QCHECK_EQ(0, cudaStreamCreate(&stream_)); in OpConverterTest()
1074 ~OpConverterTest() override { QCHECK_EQ(0, cudaStreamDestroy(stream_)); } in ~OpConverterTest()
1220 QCHECK_EQ(num_elements, values.size()) in AddTestWeights()
1225 QCHECK_EQ(weights.size_bytes(), sizeof(T) * values.size()) in AddTestWeights()
/external/tensorflow/tensorflow/core/platform/default/
Dlogging.h324 #define QCHECK_EQ(x, y) CHECK_EQ(x, y) macro