Home
last modified time | relevance | path

Searched refs:expected_output_size (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dspectrogram_test.cc124 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/
Dconstant_folding_test.cc3907 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/
Dtf_ops_a_m.cc1652 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()