Searched refs:bit_casted_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op_impl.h | 98 context->input(0).bit_casted_tensor<Proxy, NDIM>(), begin_di, sizes_di); in HandleStridedSliceCase()
111 context->input(0).bit_casted_tensor<Proxy, NDIM>(), begin_di, end_di, in HandleStridedSliceCase()
136 context->eigen_device<Device>(), result->bit_casted_tensor<Proxy, NDIM>(), in HandleStridedSliceGradCase()
157 context->eigen_device<Device>(), result->bit_casted_tensor<Proxy, NDIM>(), in operator()
|
| D | reverse_op.cc | 57 const T* in_ptr = input.bit_casted_tensor<T, 3>().data(); in ReverseRows()
58 T* out_ptr = result->bit_casted_tensor<T, 3>().data(); in ReverseRows()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 355 typename TTypes<T, NDIMS>::Tensor bit_casted_tensor();
473 typename TTypes<T, NDIMS>::ConstTensor bit_casted_tensor() const;
730 typename TTypes<T, NDIMS>::Tensor Tensor::bit_casted_tensor() { in bit_casted_tensor() function
737 typename TTypes<T, NDIMS>::ConstTensor Tensor::bit_casted_tensor() const { in bit_casted_tensor() function
|