Home
last modified time | relevance | path

Searched refs:axis_idx (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dreduce.cc111 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local
112 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()
150 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local
151 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h169 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local
170 if (idx == axis[axis_idx]) { in ReducedOutputOffset()
551 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) {
552 if (idx == axis[axis_idx]) {
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h388 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local
389 if (idx == axis[axis_idx]) { in ReducedOutputOffset()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc3766 int axis_idx = node.input_size() - 1; in GetConcatAxis() local
3767 while (axis_idx > 0 && IsControlInput(node.input(axis_idx))) { in GetConcatAxis()
3768 --axis_idx; in GetConcatAxis()
3770 if (axis_idx <= 0) { in GetConcatAxis()
3774 if (!GetTensorFromConstNode(node.input(axis_idx), &axis_tensor)) { in GetConcatAxis()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc949 int axis_idx = std::is_same<OpT, ConcatOp>() ? 0 : 1; in Verify() local
950 Value axis = *op.getODSOperands(axis_idx).begin(); in Verify()