OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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