Searched refs:stop_indices (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 49 p->stop_indices[i + pad_count] = p->stop_indices[i]; in StridedSlicePadIndices()
53 p->stop_indices[i] = 1; in StridedSlicePadIndices()
117 const auto* stop_indices = params.stop_indices; in StopForAxis() local
122 int stop = stop_indices[axis]; in StopForAxis()
171 const std::vector<int>& start_indices, const std::vector<int>& stop_indices, in BuildStridedSliceParams() argument
181 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 | 1054 int32 stop_indices[4]; 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 | 1641 if (op->start_indices.empty() || op->stop_indices.empty() || in ProcessStridedSliceOperator()
1673 CHECK_LE(op->stop_indices.size(), num_input_axes) in ProcessStridedSliceOperator()
1692 op->start_indices, op->stop_indices, op->strides); in ProcessStridedSliceOperator()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 1106 std::vector<int> stop_indices;
1121 stop_indices = other.stop_indices;
1134 CHECK_EQ(start_indices.size(), stop_indices.size());
1135 CHECK_EQ(stop_indices.size(), strides.size());
1139 stop_indices.push_back(0);
1147 CHECK_EQ(start_indices.size(), stop_indices.size());
1148 CHECK_EQ(stop_indices.size(), strides.size());
1151 std::reverse(stop_indices.begin(), stop_indices.end());
|
| D | export_tensorflow.cc | 1759 CreateSliceInput(src_op.inputs[2], src_op.stop_indices, tensorflow_graph); in ConvertStridedSliceOperator()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice.cc | 73 op_params.stop_indices[i] = GetTensorData<int32_t>(op_context->end)[i]; in BuildStridedSliceParams()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice.cc | 70 op_params.stop_indices[i] = GetTensorData<int32_t>(op_context->end)[i]; in BuildStridedSliceParams()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 478 EXPECT_EQ(op.stop_indices, output_toco_op->stop_indices); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 947 std::reverse(p->stop_indices, p->stop_indices + p->stop_indices_count); in StridedSliceReverseIndices()
969 const std::vector<int>& stop_indices, in StridedSlice() argument
974 begin_mask, end_mask, shrink_axis_mask, start_indices, stop_indices, in StridedSlice()
|