Searched refs:UnalignedTensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.h | 219 typename TTypes<T>::UnalignedTensor output(output_base, num_rows); in operator()
246 typename TTypes<T>::UnalignedTensor output(output_base, num_rows);
285 typename TTypes<T>::UnalignedTensor output(output_base, num_rows);
328 typename TTypes<T>::UnalignedTensor output(output_base, num_rows);
|
| D | quantized_concat_op.cc | 50 typename TTypes<T, 1>::UnalignedTensor output_array(dst, dims); in Copy()
|
| D | training_ops.cc | 781 auto var = typename TTypes<T>::UnalignedTensor(var_ptr + begin, t_size); in operator ()()
782 auto m = typename TTypes<T>::UnalignedTensor(m_ptr + begin, t_size); in operator ()()
783 auto v = typename TTypes<T>::UnalignedTensor(v_ptr + begin, t_size); in operator ()()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_types.h | 36 UnalignedTensor; typedef
|
| D | tensor.h | 505 typename TTypes<T, NDIMS>::UnalignedTensor unaligned_shaped(
858 typename TTypes<T, NDIMS>::UnalignedTensor Tensor::unaligned_shaped( in unaligned_shaped()
863 return typename TTypes<T, NDIMS>::UnalignedTensor(base<T>(), dims); in unaligned_shaped()
|
| D | tensor_test.cc | 450 TestReshape<TTypes<float, N>::UnalignedTensor, \ in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 72 typename TTypes<T, 1>::UnalignedTensor output_array(dst, dims); in Copy()
|