| /external/tensorflow/tensorflow/core/kernels/ |
| strided_slice_op_gpu.cu.cc | 31 template struct functor::StridedSlice<GPUDevice, T, 1>; \
32 template struct functor::StridedSlice<GPUDevice, T, 2>; \
33 template struct functor::StridedSlice<GPUDevice, T, 3>; \
34 template struct functor::StridedSlice<GPUDevice, T, 4>; \
35 template struct functor::StridedSlice<GPUDevice, T, 5>; \
36 template struct functor::StridedSlice<GPUDevice, T, 6>; \
37 template struct functor::StridedSlice<GPUDevice, T, 7>; \
|
| strided_slice_op_impl.h | 108 functor::StridedSlice<Device, Proxy, NDIM>()(
192 void StridedSlice<GPUDevice, T, NDIM>::operator()( \
198 extern template struct StridedSlice<GPUDevice, T, NDIM>; \
|
| strided_slice_op.h | 31 struct StridedSlice {
47 To32Bit(input).stridedSlice(start_i, stop_i, strides_i);
50 input.stridedSlice(start_indices, stop_indices, strides);
95 To32Bit(output).stridedSlice(start_i, stop_i, strides_i).device(d) =
98 output.stridedSlice(start_indices, stop_indices, strides).device(d) =
120 To32Bit(output).stridedSlice(start_i, stop_i, strides_i).device(d) =
123 output.stridedSlice(start_indices, stop_indices, strides).device(d) =
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| resolve_constant_strided_slice.cc | 61 void StridedSlice(StridedSliceOperator const& op, Array const& input_array,
63 // The TensorFlow documentation for StridedSlice is a bit ambiguous in places
168 StridedSlice<ArrayDataType::kFloat>(*op, input_array, &output_array);
171 StridedSlice<ArrayDataType::kUint8>(*op, input_array, &output_array);
174 StridedSlice<ArrayDataType::kInt32>(*op, input_array, &output_array);
177 StridedSlice<ArrayDataType::kInt64>(*op, input_array, &output_array);
181 << "Unsupported data type input to StridedSlice op with output \""
|
| /frameworks/ml/nn/common/operations/ |
| StridedSlice.cpp | 73 tflite::reference_ops::StridedSlice(
81 tflite::reference_ops::StridedSlice(
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| strided_slice.cc | 159 "StridedSlice op only supports 1D-4D input arrays.");
210 kernel_type::StridedSlice( \
236 "by StridedSlice.");
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| operator.cc | 622 class StridedSlice
824 ops.emplace_back(new StridedSlice(::tflite::BuiltinOperator_STRIDED_SLICE,
[all...] |
| operator_test.cc | 361 TEST_F(OperatorTest, StridedSlice) {
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| reference_ops.h | [all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| tooling_util.cc | 287 HANDLE_OPERATORTYPENAME_CASE(StridedSlice)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | [all...] |
