Searched refs:nchw_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 766 TensorShape nchw_shape = ShapeFromFormat( in operator ()() local
771 ctx->allocate_temp(DataTypeToEnum<T>::value, nchw_shape, in operator ()()
778 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in operator ()()
786 TensorShape nchw_shape = ShapeFromFormat( in operator ()() local
791 if (nchw_shape.dim_size(1) > 1) { in operator ()()
793 nchw_shape, &transformed_input)); in operator ()()
800 CHECK(transformed_input.CopyFrom(compatible_input, nchw_shape)); in operator ()()
|
| D | conv_grad_ops_3d.cc | 1302 TensorShape nchw_shape = {dims.batch_size, dims.out_depth, in Compute() local
1307 DataTypeToEnum<T>::value, nchw_shape, in Compute()
1313 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in Compute()
1695 TensorShape nchw_shape = {dims.batch_size, dims.out_depth, in Compute() local
1699 context, context->allocate_temp(DataTypeToEnum<T>::value, nchw_shape, in Compute()
1706 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in Compute()
1713 TensorShape nchw_shape = { in Compute() local
1719 nchw_shape, &transformed_input)); in Compute()
1725 CHECK(transformed_input.CopyFrom(compatible_input, nchw_shape)); in Compute()
|
| D | conv_ops_3d.cc | 329 const TensorShape nchw_shape = ShapeFromFormat( in launch() local
334 nchw_shape, &transformed_input)); in launch()
344 CHECK(input.CopyFrom(input, nchw_shape)); in launch()
|
| D | conv_ops_fused_impl.h | 648 TensorShape nchw_shape =
654 nchw_shape, &transformed_input));
662 CHECK(input.CopyFrom(input, nchw_shape)); // Crash OK
|
| D | conv_ops.cc | 743 TensorShape nchw_shape = in operator ()() local
748 nchw_shape, &transformed_input)); in operator ()()
756 CHECK(input.CopyFrom(input, nchw_shape)); in operator ()()
|
| D | conv_grad_input_ops.cc | 886 TensorShape nchw_shape = ShapeFromFormat( in operator ()() local
891 ctx->allocate_temp(DataTypeToEnum<T>::value, nchw_shape, in operator ()()
898 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 401 TensorShape nchw_shape = in TransformNHWCToNCHW() local
404 TF_RETURN_IF_ERROR(ctx->allocate_temp(DataTypeToEnum<T>::value, nchw_shape, in TransformNHWCToNCHW()
411 CHECK(transformed_tensor->CopyFrom(nhwc_tensor, nchw_shape)); in TransformNHWCToNCHW()
565 TensorShape nchw_shape = ShapeFromFormat( in launch() local
568 ctx->allocate_temp(DataTypeToEnum<T>::value, nchw_shape, in launch()
|