| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 509 inline TensorShape ShapeFromFormat(TensorFormat format, int64 N, in ShapeFromFormat() function
563 inline TensorShape ShapeFromFormat(TensorFormat format, int64 N, int64 H, in ShapeFromFormat() function
565 return ShapeFromFormat(format, N, {H, W}, C); in ShapeFromFormat()
577 inline TensorShape ShapeFromFormat(TensorFormat dst_format, in ShapeFromFormat() function
598 return ShapeFromFormat(dst_format, batch, {spatial_dims}, channels); in ShapeFromFormat()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 51 ShapeFromFormat(FORMAT_NCHW, tensor_in.shape(), in Compute()
65 ShapeFromFormat(FORMAT_NCHW, out_shape, data_format), in Compute()
137 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
148 ShapeFromFormat(FORMAT_NCHW, out_backprop.shape(), data_format); in Compute()
|
| D | pooling_ops_common.cc | 119 return ShapeFromFormat(data_format, tensor_in_batch, out_height, out_width, in forward_output_shape()
183 ShapeFromFormat(FORMAT_NCHW, tensor_in.shape(), in Compute()
196 ShapeFromFormat(FORMAT_NCHW, tensor_out_shape, in Compute()
315 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
326 ShapeFromFormat(FORMAT_NCHW, out_backprop.shape(), data_format); in Compute()
|
| D | conv_ops_3d.cc | 160 TensorShape out_shape = ShapeFromFormat( in Compute()
310 TensorShape transformed_shape = ShapeFromFormat( in launch()
329 const TensorShape nchw_shape = ShapeFromFormat( in launch()
399 ShapeFromFormat(FORMAT_NCHW, in_batch, in launch()
|
| D | fused_batch_norm_op.cc | 282 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
293 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
422 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
434 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
446 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
|
| D | conv_ops_fused_impl.h | 630 ShapeFromFormat(params.data_format, in_batch,
649 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
728 ShapeFromFormat(FORMAT_NCHW, out_batch, out_rows,
904 TensorShape out_shape = ShapeFromFormat(
|
| D | conv_ops.cc | 465 TensorShape out_shape = ShapeFromFormat( in Compute()
712 ShapeFromFormat(data_format, in_batch, new_in_rows, in operator ()()
744 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths); in operator ()()
805 ShapeFromFormat(FORMAT_NCHW, out_batch, in operator ()()
|
| D | depthtospace_op.cc | 109 ShapeFromFormat(data_format_, batch_size, output_height, in Compute()
|
| D | spacetodepth_op.cc | 110 ShapeFromFormat(data_format_, batch_size, output_height, in Compute()
|
| D | conv_grad_input_ops.cc | 823 compatible_input_shape = ShapeFromFormat( in operator ()()
886 TensorShape nchw_shape = ShapeFromFormat( in operator ()()
908 ShapeFromFormat( in operator ()()
1015 ShapeFromFormat(FORMAT_NCHW, in operator ()()
|
| D | mkl_conv_ops.h | 447 ? ShapeFromFormat(data_format_, out_batch, out_rows, out_cols,
449 : ShapeFromFormat(data_format_, out_batch,
|
| D | conv_grad_filter_ops.cc | 699 ShapeFromFormat(data_format, dims.batch_size, new_in_rows, in operator ()()
766 TensorShape nchw_shape = ShapeFromFormat( in operator ()()
786 TensorShape nchw_shape = ShapeFromFormat( in operator ()()
|
| D | conv_ops_using_gemm.cc | 528 ShapeFromFormat(data_format_, batch, out_rows, out_cols, out_depth); in Compute()
|
| D | depthwise_conv_op.cc | 356 ShapeFromFormat(data_format_, batch, out_rows, out_cols, out_depth); in Compute()
|
| D | pooling_ops_3d.cc | 96 return ShapeFromFormat(data_format, tensor_in_batch, in forward_output_shape()
190 TensorShape out_shape = ShapeFromFormat(data_format_, in_batch, in Compute()
|
| D | conv_ops_fused_image_transform.cc | 833 ShapeFromFormat(FORMAT_NHWC, batch, out_rows, out_cols, out_depth); in Compute()
|
| D | maxpooling_op.cc | 1156 ShapeFromFormat(data_format_, params.tensor_in_batch, params.out_height, in Compute()
1274 ShapeFromFormat(data_format_, params.tensor_in_batch, params.out_height, in Compute()
|
| D | conv_grad_ops_3d.cc | 1632 ShapeFromFormat(data_format_, dims.batch_size, in Compute()
|
| D | mkl_conv_ops.cc | 533 ShapeFromFormat(data_format_, batch, out_rows, out_cols, out_depth); in Compute()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 219 TensorShape output_shape = ShapeFromFormat( in Compute()
402 ShapeFromFormat(FORMAT_NCHW, batch_size, rows, cols, depth); in TransformNHWCToNCHW()
521 ShapeFromFormat(data_format, batch_size, new_conv_input_rows, in launch()
565 TensorShape nchw_shape = ShapeFromFormat( in launch()
|