Searched refs:expected_output_size (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | spectrogram_test.cc | 124 for (int expected_output_size : expected_output_sizes) { in TEST() local 126 EXPECT_EQ(expected_output_size, output.size()); in TEST() 147 for (int expected_output_size : expected_output_sizes) { in TEST() local 149 EXPECT_EQ(expected_output_size, output.size()); in TEST()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | constant_folding_test.cc | 3907 int expected_output_size = SetFetch(&item, fetch_const, fetch_cast, in ConstantFoldingCastConst() local 3912 EXPECT_EQ(expected_output_size, output.node_size()); in ConstantFoldingCastConst() 3927 int expected_output_size = 0; in SetFetch() local 3930 expected_output_size++; in SetFetch() 3934 expected_output_size++; in SetFetch() 3938 expected_output_size++; in SetFetch() 3942 expected_output_size++; in SetFetch() 3944 return expected_output_size; in SetFetch()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
D | tf_ops_a_m.cc | 1652 tensorflow::int64 expected_output_size; in inferConvReturnTypes() local 1665 get_int(dilations[dim]), stride, padding, &expected_output_size, in inferConvReturnTypes() 1669 return_shape[dim] = expected_output_size; in inferConvReturnTypes()
|