HomeSort by relevance Sort by last modified time
    Searched refs:AsEigenDSizes (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
xent_op.cc 83 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(),
89 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(),
transpose_functor.h 154 in.shape().AsEigenDSizes<NDIMS>());
157 out->shape().AsEigenDSizes<NDIMS>());
scatter_nd_op.cc 652 indices_host_, indices.shape().AsEigenDSizes<2>());
    [all...]
  /external/tensorflow/tensorflow/core/framework/
tensor_shape.h 307 Eigen::DSizes<IndexType, NDIMS> AsEigenDSizes() const;
309 /// Same as `AsEigenDSizes()` but allows for `NDIMS > dims()` -- in
467 Eigen::DSizes<IndexType, NDIMS> TensorShape::AsEigenDSizes() const {
tensor.h 719 shape().AsEigenDSizes<NDIMS>());
726 shape().AsEigenDSizes<NDIMS>());
733 shape().AsEigenDSizes<NDIMS>());
740 shape().AsEigenDSizes<NDIMS>());
    [all...]
tensor_shape_test.cc 294 Eigen::DSizes<Eigen::DenseIndex, NDIMS> AsEigenDSizes() const;
296 /// Same as `AsEigenDSizes()` but allows for `NDIMS > dims()` -- in

Completed in 743 milliseconds