Searched refs:nchw_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 1355 TensorShape nchw_shape = {dims.batch_size, dims.out_depth, in Compute() local
1360 DataTypeToEnum<T>::value, nchw_shape, in Compute()
1366 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in Compute()
1846 TensorShape nchw_shape = {dims.batch_size, dims.out_depth, in Compute() local
1850 context, context->allocate_temp(DataTypeToEnum<T>::value, nchw_shape, in Compute()
1857 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in Compute()
1865 TensorShape nchw_shape = { in Compute() local
1871 nchw_shape, &transformed_input)); in Compute()
1877 CHECK(transformed_input.CopyFrom(compatible_input, nchw_shape)); in Compute()
|
| D | conv_ops_3d.cc | 357 const TensorShape nchw_shape = ShapeFromFormat( in launch() local
362 nchw_shape, &transformed_input)); in launch()
372 CHECK(input.CopyFrom(input, nchw_shape)); in launch()
|
| D | conv_ops_fused_impl.h | 542 TensorShape nchw_shape =
548 nchw_shape, &transformed_input));
556 CHECK(input.CopyFrom(input, nchw_shape)); // Crash OK
|
| D | conv_ops.cc | 840 TensorShape nchw_shape = in operator ()() local
845 nchw_shape, &transformed_input)); in operator ()()
853 CHECK(input.CopyFrom(input, nchw_shape)); in operator ()()
|