Searched refs:compatible_input (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dconv_grad_filter_ops.cc762 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 Dconv_grad_ops_3d.cc898 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