Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlower_tf.cc1573 int32_t axis_i = axis_attr.getValues<int32_t>()[i]; in matchAndRewrite() local
1617 int32_t axis_i = adjusted_axis[i]; in matchAndRewrite()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEReverseKernel.cpp93 const int axis_i = *(reinterpret_cast<const int *>(axis->buffer()) + i); in run_reverse() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Droll.cc210 int32_t axis_i = axis_data[i]; in Eval() local
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_helper.py990 def op_wrapper(input, axis_i, keepdims_i): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1343 int64_t axis_i = axis.getInt(); in BuildGatherOp() local