| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 50 p->stop_indices[i + pad_count] = p->stop_indices[i]; in StridedSlicePadIndices()
54 p->stop_indices[i] = 1; in StridedSlicePadIndices()
127 const auto* stop_indices = params.stop_indices; in StopForAxis() local
136 int stop = stop_indices[axis]; in StopForAxis()
184 const std::vector<int>& start_indices, const std::vector<int>& stop_indices, in BuildStridedSliceParams() argument
194 op_params.stop_indices[i] = stop_indices[i]; in BuildStridedSliceParams()
|
| D | strided_slice_logic_test.cc | 36 op_params.stop_indices[i] = end.begin()[i]; in RunStridedSlicePadIndices()
44 EXPECT_EQ(op_params.stop_indices[i], expected_end.begin()[i]); in RunStridedSlicePadIndices()
|
| D | types.h | 1125 int32_t stop_indices[5]; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.h | 35 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices, in operator()
43 stop_i[i] = stop_indices[i]; in operator()
50 input.stridedSlice(start_indices, stop_indices, strides); in operator()
83 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
92 stop_i[i] = stop_indices[i];
98 output.stridedSlice(start_indices, stop_indices, strides).device(d) =
109 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
117 stop_i[i] = stop_indices[i];
123 output.stridedSlice(start_indices, stop_indices, strides).device(d) =
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_strided_slice.cc | 42 CHECK_EQ(num_input_axes, op.stop_indices.size()); in StridedSlice()
58 op.stop_indices, op.strides); in StridedSlice()
126 if (op->start_indices.empty() || op->stop_indices.empty() || in Run()
|
| D | resolve_strided_slice_attributes.cc | 119 op->stop_indices = stop_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
|
| D | propagate_fixed_sizes.cc | 1648 if (op->start_indices.empty() || op->stop_indices.empty() || in ProcessStridedSliceOperator()
1680 CHECK_LE(op->stop_indices.size(), num_input_axes) in ProcessStridedSliceOperator()
1699 op->start_indices, op->stop_indices, op->strides); in ProcessStridedSliceOperator()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 1109 std::vector<int> stop_indices;
1124 stop_indices = other.stop_indices;
1137 CHECK_EQ(start_indices.size(), stop_indices.size());
1138 CHECK_EQ(stop_indices.size(), strides.size());
1142 stop_indices.push_back(0);
1150 CHECK_EQ(start_indices.size(), stop_indices.size());
1151 CHECK_EQ(stop_indices.size(), strides.size());
1154 std::reverse(stop_indices.begin(), stop_indices.end());
|
| D | export_tensorflow.cc | 1785 CreateSliceInput(src_op.inputs[2], src_op.stop_indices, tensorflow_graph); in ConvertStridedSliceOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice.cc | 84 op_params.stop_indices[i] = GetTensorData<int32_t>(op_context->end)[i]; in BuildStridedSliceParams()
91 op_params.start_indices[i] = op_params.stop_indices[i] = 0; in BuildStridedSliceParams()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice.cc | 70 op_params.stop_indices[i] = GetTensorData<int32_t>(op_context->end)[i]; in BuildStridedSliceParams()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 129 int32_t stop_indices[5]; member
1222 int32_t stop_indices[5];
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 492 EXPECT_EQ(op.stop_indices, output_toco_op->stop_indices); in TEST_F()
981 op.stop_indices = {1, 2, 2, 2, 2}; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 917 std::reverse(p->stop_indices, p->stop_indices + p->stop_indices_count); in StridedSliceReverseIndices()
939 const std::vector<int>& stop_indices, in StridedSlice() argument
944 begin_mask, end_mask, shrink_axis_mask, start_indices, stop_indices, in StridedSlice()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4955 const std::vector<int>& stop_indices, in StridedSlice() argument
4960 begin_mask, end_mask, shrink_axis_mask, start_indices, stop_indices, in StridedSlice()
|