HomeSort by relevance Sort by last modified time
    Searched defs:Dims (Results 1 - 6 of 6) 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/core/kernels/
mirror_pad_op.h 95 static constexpr int Dims = internal::array_size<PaddingDimensions>::value;
96 typedef DSizes<Index, Dims> Dimensions;
115 EIGEN_STATIC_ASSERT(Dims > 0, YOU_MADE_A_PROGRAMMING_MISTAKE)
126 for (int dim = 0; dim < Dims; ++dim) {
136 for (int i = 0; i < Dims - 1; ++i) {
141 input_strides_[numext::maxi(0, Dims - 1)] = 1;
142 output_strides_[numext::maxi(0, Dims - 1)] = 1;
143 for (int i = Dims - 1; i > 0; --i) {
169 coeff(array<Index, Dims> coords) const {
170 for (int dim = 0; dim < Dims; ++dim)
    [all...]
  /external/swiftshader/third_party/llvm-7.0/llvm/lib/Target/AMDGPU/
AMDGPUHSAMetadataStreamer.cpp 194 std::vector<uint32_t> Dims;
196 return Dims;
199 Dims.push_back(mdconst::extract<ConstantInt>(Op)->getZExtValue());
200 return Dims;
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/
types.h 117 struct Dims {
174 inline int32 Dims(int i) const {
228 // (a) as Dims<4>-dependent code is eliminated, the reliance on this should be
279 // Converts inference-style shape to legacy tflite::Dims<4>.
280 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) {
281 tflite::Dims<4> result;
287 (i < dimensions_count) ? array_shape.Dims(dimensions_count - 1 - i) : 1;
296 inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) {
298 {dims.sizes[3], dims.sizes[2], dims.sizes[1], dims.sizes[0]})
    [all...]
  /external/tensorflow/tensorflow/lite/kernels/internal/
types.h 116 struct Dims {
181 inline int32 Dims(int i) const {
243 // (a) as Dims<4>-dependent code is eliminated, the reliance on this should be
295 // Converts inference-style shape to legacy tflite::Dims<4>.
296 inline tflite::Dims<4> ToRuntimeDims(const tflite::RuntimeShape& array_shape) {
297 tflite::Dims<4> result;
303 (i < dimensions_count) ? array_shape.Dims(dimensions_count - 1 - i) : 1;
312 inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) {
314 {dims.sizes[3], dims.sizes[2], dims.sizes[1], dims.sizes[0]})
    [all...]

Completed in 359 milliseconds