HomeSort by relevance Sort by last modified time
    Searched defs:Dims (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/neon/
types.h 26 struct Dims {
31 inline int Offset(const Dims<4>& dims, int i0, int i1, int i2, int i3) {
32 DCHECK(i0 >= 0 && i0 < dims.sizes[0]);
33 DCHECK(i1 >= 0 && i1 < dims.sizes[1]);
34 DCHECK(i2 >= 0 && i2 < dims.sizes[2]);
35 DCHECK(i3 >= 0 && i3 < dims.sizes[3]);
36 return i0 * dims.strides[0] + i1 * dims.strides[1] + i2 * dims.strides[2]
    [all...]
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorReductionSycl.h 142 typedef typename Eigen::internal::remove_all<decltype(self.xprDims())>::type Dims;
143 Dims dims= self.xprDims(); local
156 const auto device_self_expr= TensorReductionOp<Op, Dims, decltype(device_expr.expr) ,MakeGlobalPointer>(device_expr.expr, dims, functor);
204 typedef typename Eigen::internal::remove_all<decltype(self.xprDims())>::type Dims;
205 Dims dims= self.xprDims(); local
219 const auto device_self_expr= TensorReductionOp<Op, Dims, decltype(device_expr.expr) ,MakeGlobalPointer>(device_expr.expr, dims, functor)
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/
types.h 25 struct Dims {
31 inline bool NextIndex(const int num_dims, const int* dims, int* current) {
33 TFLITE_DCHECK(dims != nullptr);
38 TFLITE_DCHECK_GE(dims[idx], current_val);
39 if (dims[idx] == current_val) {
55 // TODO(kanlig): uses Dims to represent dimensions.
56 inline size_t ReducedOutputOffset(const int num_dims, const int* dims,
60 TFLITE_DCHECK(dims != nullptr);
75 offset = offset * static_cast<size_t>(dims[idx]) +
82 inline int Offset(const Dims<4>& dims, int i0, int i1, int i2, int i3)
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
mirror_pad_op.h 95 static constexpr int Dims = internal::array_size<PaddingDimensions>::value;
96 typedef DSizes<Index, Dims> Dimensions;
114 EIGEN_STATIC_ASSERT(Dims > 0, YOU_MADE_A_PROGRAMMING_MISTAKE)
125 for (int dim = 0; dim < Dims; ++dim) {
135 for (int i = 0; i < Dims - 1; ++i) {
140 input_strides_[numext::maxi(0, Dims - 1)] = 1;
141 output_strides_[numext::maxi(0, Dims - 1)] = 1;
142 for (int i = Dims - 1; i > 0; --i) {
168 coeff(array<Index, Dims> coords) const {
169 for (int dim = 0; dim < Dims; ++dim)
    [all...]

Completed in 572 milliseconds