Searched refs:DescribeTensor4D (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 148 void DescribeTensor4D(int dim0, int dim1, int dim2, int dim3, in DescribeTensor4D() function
167 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeConvolution()
168 DescribeTensor4D(kx, ky, iz2, oz, op_context.op_info.add_inputs()); in DescribeConvolution()
184 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeDepthwiseConv2dNative()
185 DescribeTensor4D(kx, ky, iz2, cm, op_context.op_info.add_inputs()); in DescribeDepthwiseConv2dNative()
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()
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()
228 DescribeTensor4D(batch, ox, oy, oz, side_input); in DescribeFusedConv2DBiasActivation()
[all …]
|