Searched refs:compatible_input (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | conv_grad_filter_ops.cc | 762 Tensor compatible_input; local 774 &compatible_input)); 779 To32Bit(compatible_input.tensor<T, 4>()), data_format); 781 compatible_input = input; 789 .set_height(GetTensorDim(compatible_input, data_format, 'H')) 790 .set_width(GetTensorDim(compatible_input, data_format, 'W')) 857 FORMAT_NCHW, GetTensorDim(compatible_input, data_format, 'N'), 858 GetTensorDim(compatible_input, data_format, 'H'), 859 GetTensorDim(compatible_input, data_format, 'W'), 860 GetTensorDim(compatible_input, data_forma [all...] |
H A D | conv_grad_ops_3d.cc | 898 Tensor compatible_input; variable 907 &compatible_input)); 912 To32Bit(compatible_input.tensor<T, 5>()), data_format_); 914 compatible_input = input; 923 GetTensorDim(compatible_input, data_format_, '2')) 925 GetTensorDim(compatible_input, data_format_, '1')) 927 GetTensorDim(compatible_input, data_format_, '0')) 978 TensorShape nchw_shape = {batch, in_depth, compatible_input.dim_size(1), 979 compatible_input.dim_size(2), 980 compatible_input [all...] |
Completed in 129 milliseconds