Home
last modified time | relevance | path

Searched refs:compatible_input_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc198 TensorShape compatible_input_shape; in operator ()() local
209 compatible_input_shape = ShapeFromFormat( in operator ()()
212 compatible_input_shape = input_shape; in operator ()()
251 .set_height(GetTensorDim(compatible_input_shape, data_format, 'H')) in operator ()()
252 .set_width(GetTensorDim(compatible_input_shape, data_format, 'W')) in operator ()()
340 GetTensorDim(compatible_input_shape, data_format, 'N'), in operator ()()
341 GetTensorDim(compatible_input_shape, data_format, 'H'), in operator ()()
342 GetTensorDim(compatible_input_shape, data_format, 'W'), in operator ()()
343 GetTensorDim(compatible_input_shape, data_format, 'C')), in operator ()()
Dconv_grad_ops_3d.cc1250 TensorShape compatible_input_shape; in Compute() local
1253 compatible_input_shape = { in Compute()
1261 compatible_input_shape = {dims.batch_size, dims.in_depth, in Compute()
1300 .set_spatial_dim(DimIndex::X, compatible_input_shape.dim_size(4)) in Compute()
1301 .set_spatial_dim(DimIndex::Y, compatible_input_shape.dim_size(3)) in Compute()
1302 .set_spatial_dim(DimIndex::Z, compatible_input_shape.dim_size(2)) in Compute()
1377 compatible_input_shape.dim_size(0), in Compute()
1378 {{compatible_input_shape.dim_size(2), in Compute()
1379 compatible_input_shape.dim_size(3), in Compute()
1380 compatible_input_shape.dim_size(4)}}, in Compute()
[all …]