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