Searched refs:iz2 (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator_test.cc | 161 OpContext DescribeConvolution(int batch, int ix, int iy, int iz1, int iz2, in DescribeConvolution() argument 168 DescribeTensor4D(kx, ky, iz2, oz, op_context.op_info.add_inputs()); in DescribeConvolution() 179 int iz2, int kx, int ky, int cm) { in DescribeDepthwiseConv2dNative() argument 185 DescribeTensor4D(kx, ky, iz2, cm, op_context.op_info.add_inputs()); in DescribeDepthwiseConv2dNative() 199 int iz2, int kx, int ky, int ox, in DescribeFusedConv2DBiasActivation() argument 217 DescribeTensor4D(kx, ky, iz2, oz, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation() 220 DescribeTensor4D(oz, iz2, kx, ky, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
|