Home
last modified time | relevance | path

Searched defs:DenseIndex (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dsharding_util_ops.cc180 const Eigen::DSizes<Eigen::DenseIndex, Rank>& slice_shape, in GetSliceIndices()
188 const Eigen::DSizes<Eigen::DenseIndex, 1>& slice_shape, const int index) { in GetSliceIndices()
197 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_shape, const int index) { in GetSliceIndices()
207 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_shape, const int index) { in GetSliceIndices()
220 const Eigen::DSizes<Eigen::DenseIndex, 4>& slice_shape, const int index) { in GetSliceIndices()
237 const Eigen::DSizes<Eigen::DenseIndex, 5>& slice_shape, const int index) { in GetSliceIndices()
258 const Eigen::DSizes<Eigen::DenseIndex, 6>& slice_shape, const int index) { in GetSliceIndices()
284 const Eigen::DSizes<Eigen::DenseIndex, 7>& slice_shape, const int index) { in GetSliceIndices()
315 const Eigen::DSizes<Eigen::DenseIndex, 8>& slice_shape, const int index) { in GetSliceIndices()
550 const Eigen::DSizes<Eigen::DenseIndex, Rank>& slice_indices, in AssignFromInput()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsplit_lib_gpu.cu.cc36 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices, in operator ()()
45 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_indices, in operator ()()
H A Dsplit_lib_cpu.cc31 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices, in operator ()()
H A Dgather_nd_op_cpu_impl.h55 const Index loc, Eigen::array<Eigen::DenseIndex, IXDIM + 1>* ix) const { in GenerateIndices()
H A Dscatter_nd_op_gpu.cu.cc100 const Eigen::array<Eigen::DenseIndex, IXDIM> output_shape_prefix, in ScatterNdOpKernel()
H A Ddata_format_ops.cc224 Eigen::DSizes<Eigen::DenseIndex, 10>* dst) { in ComputeDstIndex()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor_slice.h207 const TensorShape& shape, Eigen::DSizes<Eigen::DenseIndex, NDIMS>* indices, in FillIndicesAndSizes()
H A Dtensor.h825 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const { in FillDimsAndValidateCompatibleShape()
838 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const { in FillDimsAndValidateCompatibleShape()
/aosp_15_r20/external/tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/
H A DMatMatProductAVX2.h91 gemm_blocking_space(DenseIndex rows, DenseIndex cols, DenseIndex depth, in gemm_blocking_space() argument
/aosp_15_r20/external/eigen/Eigen/src/Core/util/
H A DMeta.h66 typedef EIGEN_DEFAULT_DENSE_INDEX_TYPE DenseIndex; typedef
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorRef.h69 EIGEN_DEVICE_FUNC virtual Scalar& coeffRef(DenseIndex /*index*/) { in coeffRef() argument