Searched refs:reinterpret_last_dimension (Results 1 – 4 of 4) 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 | 133 auto Tinput_v = input.template reinterpret_last_dimension<int32, 4>(); in Compute()
134 auto Toutput_v = outputs_tensor->reinterpret_last_dimension<int32, 4>(); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 428 typename TTypes<T, NDIMS>::Tensor reinterpret_last_dimension();
546 typename TTypes<T, NDIMS>::ConstTensor reinterpret_last_dimension() const;
772 typename TTypes<T, NDIMS>::Tensor Tensor::reinterpret_last_dimension() { in reinterpret_last_dimension() function
787 typename TTypes<T, NDIMS>::ConstTensor Tensor::reinterpret_last_dimension() in reinterpret_last_dimension() function
|
| D | tensor_test.cc | 775 auto actual_nchw = t_nchw_vect_c.reinterpret_last_dimension<int32, 4>(); in TEST()
778 const_t_nchw_vect_c.reinterpret_last_dimension<int32, 4>(); in TEST()
|