| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 522 inline TensorShape ShapeFromFormat(TensorFormat format, int64 N, in ShapeFromFormat() function
576 inline TensorShape ShapeFromFormat(TensorFormat format, int64 N, int64 H, in ShapeFromFormat() function
578 return ShapeFromFormat(format, N, {H, W}, C); in ShapeFromFormat()
590 inline TensorShape ShapeFromFormat(TensorFormat dst_format, in ShapeFromFormat() function
611 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()
152 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
163 ShapeFromFormat(FORMAT_NCHW, out_backprop.shape(), data_format); in Compute()
|
| D | pooling_ops_common.cc | 201 return ShapeFromFormat(data_format, tensor_in_batch, out_height, out_width, in forward_output_shape()
247 ShapeFromFormat(FORMAT_NCHW, tensor_in.shape(), in Compute()
260 ShapeFromFormat(FORMAT_NCHW, tensor_out_shape, in Compute()
317 ShapeFromFormat(data_format, batch_size, in Compute()
475 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
486 ShapeFromFormat(FORMAT_NCHW, out_backprop.shape(), data_format); in Compute()
616 ShapeFromFormat(transformed_input_data_format, batch_size, in Compute()
623 ShapeFromFormat(transformed_input_data_format, batch_size, in Compute()
|
| D | fused_batch_norm_op.cc | 128 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
133 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
266 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
271 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
358 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
363 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
368 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
864 ShapeFromFormat(compute_format, batch_size, in operator ()()
875 ShapeFromFormat(compute_format, batch_size, in operator ()()
1041 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
[all …]
|
| D | lrn_op.cc | 243 ShapeFromFormat(FORMAT_NCHW, in.shape(), FORMAT_NHWC), in launch()
253 ShapeFromFormat(FORMAT_NCHW, output->shape(), FORMAT_NHWC), in launch()
536 ShapeFromFormat(FORMAT_NCHW, in_grads.shape(), in launch()
546 ShapeFromFormat(FORMAT_NCHW, in_image.shape(), in launch()
556 ShapeFromFormat(FORMAT_NCHW, out_image.shape(), in launch()
567 ShapeFromFormat(FORMAT_NCHW, output->shape(), FORMAT_NHWC), in launch()
|
| D | conv_grad_input_ops.cc | 209 compatible_input_shape = ShapeFromFormat( in operator ()()
316 TensorShape compute_shape = ShapeFromFormat( in operator ()()
338 ShapeFromFormat( in operator ()()
515 ShapeFromFormat(compute_data_format, in operator ()()
|
| D | conv_ops_3d.cc | 169 TensorShape out_shape = ShapeFromFormat( in Compute()
322 TensorShape transformed_shape = ShapeFromFormat( in launch()
357 const TensorShape nchw_shape = ShapeFromFormat( in launch()
460 ShapeFromFormat(FORMAT_NCHW, in_batch, in launch()
|
| D | conv_ops_fused_impl.h | 508 ShapeFromFormat(params.data_format, in_batch,
543 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
622 ShapeFromFormat(FORMAT_NCHW, out_batch, out_rows,
770 TensorShape out_shape = ShapeFromFormat(
|
| D | depthtospace_op.cc | 109 ShapeFromFormat(data_format_, batch_size, output_height, in Compute()
|
| D | conv_grad_shape_utils.cc | 189 *input_shape = ShapeFromFormat(data_format, batch_size, output_height, in Conv2DBackpropComputeInputShape()
|
| D | spacetodepth_op.cc | 125 ShapeFromFormat(data_format_, batch_size, output_height, in Compute()
|
| D | conv_ops.cc | 531 TensorShape out_shape = ShapeFromFormat( in Compute()
808 ShapeFromFormat(data_format, in_batch, new_in_rows, in operator ()()
841 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths); in operator ()()
944 ShapeFromFormat(compute_data_format, out_batch, in operator ()()
|
| D | conv_grad_filter_ops.cc | 806 ShapeFromFormat(data_format, dims.batch_size, new_in_rows, in operator ()()
901 TensorShape compute_shape = ShapeFromFormat( in operator ()()
922 TensorShape compute_shape = ShapeFromFormat( in operator ()()
|
| D | conv_ops_using_gemm.cc | 530 ShapeFromFormat(data_format_, batch, out_rows, out_cols, out_depth); in Compute()
|
| D | depthwise_conv_op.cc | 389 ShapeFromFormat(data_format_, batch, out_rows, out_cols, out_depth); in Compute()
|
| D | pooling_ops_3d.cc | 91 return ShapeFromFormat(data_format, tensor_in_batch, in forward_output_shape()
185 TensorShape out_shape = ShapeFromFormat(data_format_, in_batch, in Compute()
|
| D | conv_grad_ops_3d.cc | 1376 ShapeFromFormat(compute_data_format, in Compute()
1537 ShapeFromFormat(compute_data_format, dims.batch_size, in Compute()
1745 ShapeFromFormat(data_format_, dims.batch_size, in Compute()
|
| D | conv_ops_fused_image_transform.cc | 830 ShapeFromFormat(FORMAT_NHWC, batch, out_rows, out_cols, out_depth); in Compute()
|
| D | maxpooling_op.cc | 1201 ShapeFromFormat(data_format_, params.tensor_in_batch, params.out_height, in Compute()
1329 ShapeFromFormat(data_format_, params.tensor_in_batch, params.out_height, in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 456 ? ShapeFromFormat(data_format_, out_batch, out_rows, out_cols,
458 : ShapeFromFormat(data_format_, out_batch,
|