Searched refs:compatible_input_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 812 TensorShape compatible_input_shape; in operator ()() local
823 compatible_input_shape = ShapeFromFormat( in operator ()()
826 compatible_input_shape = input_shape; in operator ()()
834 .set_height(GetTensorDim(compatible_input_shape, data_format, 'H')) in operator ()()
835 .set_width(GetTensorDim(compatible_input_shape, data_format, 'W')) in operator ()()
910 GetTensorDim(compatible_input_shape, data_format, 'N'), in operator ()()
911 GetTensorDim(compatible_input_shape, data_format, 'H'), in operator ()()
912 GetTensorDim(compatible_input_shape, data_format, 'W'), in operator ()()
913 GetTensorDim(compatible_input_shape, data_format, 'C')), in operator ()()
|
| D | conv_grad_ops_3d.cc | 1235 TensorShape compatible_input_shape; in Compute() local
1238 compatible_input_shape = { in Compute()
1246 compatible_input_shape = {dims.batch_size, dims.in_depth, in Compute()
1256 .set_spatial_dim(DimIndex::X, compatible_input_shape.dim_size(4)) in Compute()
1257 .set_spatial_dim(DimIndex::Y, compatible_input_shape.dim_size(3)) in Compute()
1258 .set_spatial_dim(DimIndex::Z, compatible_input_shape.dim_size(2)) in Compute()
1322 context->allocate_temp(DataTypeToEnum<T>::value, compatible_input_shape, in Compute()
|