| D | op_level_cost_estimator_test.cc | 161 OpContext DescribeConvolution(int batch, int ix, int iy, int iz1, int iz2, in DescribeConvolution() argument
167 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeConvolution()
178 OpContext DescribeDepthwiseConv2dNative(int batch, int ix, int iy, int iz1, in DescribeDepthwiseConv2dNative() argument
184 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeDepthwiseConv2dNative()
198 OpContext DescribeFusedConv2DBiasActivation(int batch, int ix, int iy, int iz1, in DescribeFusedConv2DBiasActivation() argument
211 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
214 DescribeTensor4D(batch, iz1, ix, iy, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
|