Home
last modified time | relevance | path

Searched defs:axis_i (Results 1 – 4 of 4) sorted by relevance

/external/ComputeLibrary/src/core/NEON/kernels/
DNEReverseKernel.cpp93 const int axis_i = *(reinterpret_cast<const int *>(axis->buffer()) + i); in run_reverse() local
/external/tensorflow/tensorflow/lite/kernels/
Droll.cc210 int32_t axis_i = axis_data[i]; in Eval() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1581 int32_t axis_i = axis_attr.getValue<int32_t>(i); in matchAndRewrite() local
1625 int32_t axis_i = adjusted_axis[i]; in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1201 int64_t axis_i = axis.getInt(); in BuildGatherOp() local