Searched refs:AsEigenDSizes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xent_op.cc | 80 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(), in Compute()
86 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(), in Compute()
|
| D | transpose_functor.h | 154 in.shape().AsEigenDSizes<NDIMS>()); in TransposeUsingEigen()
157 out->shape().AsEigenDSizes<NDIMS>()); in TransposeUsingEigen()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 747 shape().AsEigenDSizes<NDIMS>()); in tensor()
754 shape().AsEigenDSizes<NDIMS>()); in tensor()
761 shape().AsEigenDSizes<NDIMS>()); in bit_casted_tensor()
768 shape().AsEigenDSizes<NDIMS>()); in bit_casted_tensor()
|
| D | tensor_shape.h | 375 Eigen::DSizes<IndexType, NDIMS> AsEigenDSizes() const;
604 Eigen::DSizes<IndexType, NDIMS> TensorShape::AsEigenDSizes() const { in AsEigenDSizes() function
|
| D | tensor_shape_test.cc | 317 Eigen::DSizes<Eigen::DenseIndex, NDIMS> AsEigenDSizes() const;
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 409 device, input_map, TensorShape(strides).AsEigenDSizes<N>(), \ in StrideOrInflate()
415 device, input_map, TensorShape(strides).AsEigenDSizes<N>(), \ in StrideOrInflate()
|