Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc161 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()