Searched refs:UnalignedTensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.h | 220 typename TTypes<T>::UnalignedTensor output(output_base, num_rows); in operator()
247 typename TTypes<T>::UnalignedTensor output(output_base, num_rows);
286 typename TTypes<T>::UnalignedTensor output(output_base, num_rows);
329 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 | 809 auto var = typename TTypes<T>::UnalignedTensor(var_ptr + begin, t_size); in operator ()()
810 auto m = typename TTypes<T>::UnalignedTensor(m_ptr + begin, t_size); in operator ()()
811 auto v = typename TTypes<T>::UnalignedTensor(v_ptr + begin, t_size); in operator ()()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_types.h | 37 UnalignedTensor; typedef
|
| D | tensor.h | 518 typename TTypes<T, NDIMS>::UnalignedTensor unaligned_shaped(
897 typename TTypes<T, NDIMS>::UnalignedTensor Tensor::unaligned_shaped( in unaligned_shaped()
902 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 | 76 typename TTypes<T, 1>::UnalignedTensor output_array(dst, dims); in Copy()
|