Searched refs:reinterpret_last_dimension (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthtospace_op.cc | 118 auto Tinput_v = input.template reinterpret_last_dimension<int32, 4>(); in Compute()
119 auto Toutput_v = outputs_tensor->reinterpret_last_dimension<int32, 4>(); in Compute()
|
| D | spacetodepth_op.cc | 120 auto Tinput_v = input.template reinterpret_last_dimension<int32, 4>(); in Compute()
121 auto Toutput_v = outputs_tensor->reinterpret_last_dimension<int32, 4>(); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 365 typename TTypes<T, NDIMS>::Tensor reinterpret_last_dimension();
483 typename TTypes<T, NDIMS>::ConstTensor reinterpret_last_dimension() const;
744 typename TTypes<T, NDIMS>::Tensor Tensor::reinterpret_last_dimension() { in reinterpret_last_dimension() function
759 typename TTypes<T, NDIMS>::ConstTensor Tensor::reinterpret_last_dimension() in reinterpret_last_dimension() function
|
| D | tensor_test.cc | 742 auto actual_nchw = t_nchw_vect_c.reinterpret_last_dimension<int32, 4>(); in TEST()
745 const_t_nchw_vect_c.reinterpret_last_dimension<int32, 4>(); in TEST()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 526 To32Bit(conv_input_param.reinterpret_last_dimension<VectT, 4>()); in launch()
528 maybe_padded_conv_input.reinterpret_last_dimension<VectT, 4>()); in launch()
|