Home
last modified time | relevance | path

Searched refs:num_input_axes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_strided_slice_attributes.cc82 int num_input_axes = input_array.shape().dimensions_count(); in Run() local
95 CHECK_LE(start_indices_size, num_input_axes) in Run()
96 << "StridedSlice op requires no more than " << num_input_axes in Run()
98 CHECK_LE(stop_indices_size, num_input_axes) in Run()
99 << "StridedSlice op requires no more than " << num_input_axes in Run()
101 CHECK_LE(stride_indices_size, num_input_axes) in Run()
102 << "StridedSlice op requires no more than " << num_input_axes in Run()
110 std::vector<int> begin_pad_values(num_input_axes, 0); in Run()
115 std::vector<int> stride_pad_values(num_input_axes, 1); in Run()
Dresolve_constant_strided_slice.cc40 int num_input_axes = op.start_indices.size(); in StridedSlice() local
41 CHECK_EQ(num_input_axes, op.start_indices.size()); in StridedSlice()
42 CHECK_EQ(num_input_axes, op.stop_indices.size()); in StridedSlice()
43 CHECK_EQ(num_input_axes, op.strides.size()); in StridedSlice()
53 std::vector<int> src_coord(num_input_axes); in StridedSlice()
54 std::vector<int> stop_for_axis(num_input_axes); in StridedSlice()
60 for (int axis = 0; axis < num_input_axes; axis++) { in StridedSlice()
81 for (int axis = num_input_axes - 1; axis >= 0; --axis) { in StridedSlice()
Dpropagate_fixed_sizes.cc1669 int num_input_axes = input_array.shape().dimensions_count(); in ProcessStridedSliceOperator() local
1670 CHECK_LE(op->start_indices.size(), num_input_axes) in ProcessStridedSliceOperator()
1672 << "\", requires no more than " << num_input_axes << " start indices"; in ProcessStridedSliceOperator()
1673 CHECK_LE(op->stop_indices.size(), num_input_axes) in ProcessStridedSliceOperator()
1675 << "\", requires no more than " << num_input_axes << " stop indices"; in ProcessStridedSliceOperator()
1676 CHECK_LE(op->strides.size(), num_input_axes) in ProcessStridedSliceOperator()
1678 << "\", requires no more than " << num_input_axes << " strides"; in ProcessStridedSliceOperator()
1688 for (int axis = 0; axis < num_input_axes; ++axis) { in ProcessStridedSliceOperator()