Searched refs:kTensorSliceMaxRank (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_util.h | 31 static const int kTensorSliceMaxRank = 8; variable
36 Eigen::TensorMap<Eigen::Tensor<T, kTensorSliceMaxRank, Eigen::RowMajor>>
38 Eigen::DSizes<Eigen::DenseIndex, kTensorSliceMaxRank> dsizes = in GetEigenTensorMapFromTensorShape()
39 shape.AsEigenDSizesWithPadding<kTensorSliceMaxRank>(); in GetEigenTensorMapFromTensorShape()
40 Eigen::TensorMap<Eigen::Tensor<T, kTensorSliceMaxRank, Eigen::RowMajor>> eig( in GetEigenTensorMapFromTensorShape()
63 static_assert(kTensorSliceMaxRank == 8,
100 static_assert(kTensorSliceMaxRank == 8,
142 CHECK_LE(shape.dims(), kTensorSliceMaxRank)
143 << "Only tensors of size up to " << kTensorSliceMaxRank
176 Eigen::DSizes<Eigen::DenseIndex, kTensorSliceMaxRank> s_start, s_len,
[all …]
|