| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op_gpu_impl.h | 34 template struct functor::StridedSlice<GPUDevice, T, 1>; \
35 template struct functor::StridedSlice<GPUDevice, T, 2>; \
36 template struct functor::StridedSlice<GPUDevice, T, 3>; \
37 template struct functor::StridedSlice<GPUDevice, T, 4>; \
38 template struct functor::StridedSlice<GPUDevice, T, 5>; \
39 template struct functor::StridedSlice<GPUDevice, T, 6>; \
40 template struct functor::StridedSlice<GPUDevice, T, 7>; \
|
| D | strided_slice_op_impl.h | 108 functor::StridedSlice<Device, Proxy, NDIM>()( in HandleStridedSliceCase()
191 void StridedSlice<GPUDevice, T, NDIM>::operator()( \
197 extern template struct StridedSlice<GPUDevice, T, NDIM>; \
|
| D | strided_slice_op.h | 31 struct StridedSlice { struct
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_strided_slice.cc | 28 void StridedSlice(StridedSliceOperator const& op, Array const& input_array, in StridedSlice() function
145 StridedSlice<ArrayDataType::kFloat>(*op, input_array, &output_array); in Run()
148 StridedSlice<ArrayDataType::kUint8>(*op, input_array, &output_array); in Run()
151 StridedSlice<ArrayDataType::kInt32>(*op, input_array, &output_array); in Run()
154 StridedSlice<ArrayDataType::kInt64>(*op, input_array, &output_array); in Run()
157 StridedSlice<ArrayDataType::kComplex64>(*op, input_array, &output_array); in Run()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_StridedSliceGrad.pbtxt | 3 summary: "Returns the gradient of `StridedSlice`."
5 Since `StridedSlice` cuts out pieces of its `input` which is size
12 shape of `StridedSlice`'s `input`.
|
| D | api_def_StridedSliceAssign.pbtxt | 7 `begin`, `end`, `strides`, etc. work exactly as in `StridedSlice`.
|
| D | api_def_ResourceStridedSliceAssign.pbtxt | 7 `begin, `end`, `strides`, etc. work exactly as in `StridedSlice`.
|
| D | api_def_StridedSlice.pbtxt | 2 graph_op_name: "StridedSlice"
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | cwise_op_vectorizer.cc | 90 Output concat_pre = ops::StridedSlice( in ExpandDimsForBroadcast()
92 ops::StridedSlice::Attrs().BeginMask(1)); in ExpandDimsForBroadcast()
95 Output concat_post = ops::StridedSlice( in ExpandDimsForBroadcast()
97 ops::StridedSlice::Attrs().EndMask(1)); in ExpandDimsForBroadcast()
|
| D | reshape_vectorizer.cc | 40 ops::StridedSlice(*s, shape, const_vec_1, const_vec_1, const_vec_1, in GetVectorizedShape()
41 ops::StridedSlice::Attrs().BeginMask(1)); in GetVectorizedShape()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 370 auto y = StridedSlice(scope_, x, {2, 1, 1}, {6, 3, 3}, {2, 1, 1}); in TEST_F()
377 y = StridedSlice(scope_, x, {2, 1, 1}, {6, 3, 3}, {2, 1, 1}, in TEST_F()
378 StridedSlice::BeginMask(1 << 1).EndMask(1 << 2)); in TEST_F()
383 y = StridedSlice(scope_, x, {0, 2, 1, 1}, {0, 6, 3, 3}, {1, 2, 1, 1}, in TEST_F()
384 StridedSlice::NewAxisMask(1 << 0)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_StridedSlice.pbtxt | 2 graph_op_name: "StridedSlice"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_StridedSlice.pbtxt | 2 graph_op_name: "StridedSlice"
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 2566 op: "StridedSlice" in CreateImagenetDatasetFunctions()
2691 op: "StridedSlice" in CreateImagenetDatasetFunctions()
2873 op: "StridedSlice" in CreateImagenetDatasetFunctions()
3018 op: "StridedSlice" in CreateImagenetDatasetFunctions()
3252 op: "StridedSlice" in CreateImagenetDatasetFunctions()
3397 op: "StridedSlice" in CreateImagenetDatasetFunctions()
3690 op: "StridedSlice" in CreateImagenetDatasetFunctions()
3832 op: "StridedSlice" in CreateImagenetDatasetFunctions()
3974 op: "StridedSlice" in CreateImagenetDatasetFunctions()
4511 op: "StridedSlice" in CreateImagenetDatasetFunctions()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice.cc | 211 kernel_type::StridedSlice(op_params, GetTensorShape(op_context.input), \ in Eval()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1744 Output b = ops::StridedSlice(s.WithOpName("b"), shp, index1, index2, index2); in TEST_F()
1745 Output c = ops::StridedSlice(s.WithOpName("c"), shp, index2, index3, index2); in TEST_F()
1778 ops::StridedSlice(scope.WithOpName("slice"), input_shape, begin, end, in TEST_F()
1779 stride, ops::StridedSlice::ShrinkAxisMask(1)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 75 auto slice = ops::StridedSlice(root, input, begin_op, end_op, stride_op, in TestStridedSlice()
76 ops::StridedSlice::BeginMask(begin_mask) in TestStridedSlice()
1241 auto slice = ops::StridedSlice(root, input, begin, end, stride); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 2064 ops::StridedSlice s1(scope.WithOpName("s1"), in1, begin, end, strides); in TEST_F()
2065 ops::StridedSlice s2(scope.WithOpName("s2"), in2, begin, end, strides); in TEST_F()
2113 ops::StridedSlice s1( in TEST_F()
2115 ops::StridedSlice::Attrs().BeginMask(1).EndMask(1).EllipsisMask(2)); in TEST_F()
2123 ops::StridedSlice s2(scope.WithOpName("s2"), in2, begin2, end2, strides2); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | cpu_supported_ops.md | 240 `StridedSlice` | `Index={int32,int64}`<br>`T={bool,complex64,double,float,in…
|
| D | gpu_supported_ops.md | 237 `StridedSlice` | `Index={int32,int64}`<br>`T={bool,complex64,double,float,in…
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 454 TEST_F(OperatorTest, StridedSlice) { in TEST_F() argument
|
| D | operator.cc | 1475 class StridedSlice class
2423 ops.push_back(MakeUnique<StridedSlice>( in BuildOperatorList()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2624 ops::StridedSlice::Attrs attrs = ops::StridedSlice::Attrs() in TEST_F()
2630 auto strided_slice = ops::StridedSlice(s.WithOpName("my_strided_slice"), in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 845 inline void StridedSlice(const T* input_data, const Dims<4>& input_dims, in StridedSlice() function
857 StridedSlice(op_params, DimsToShape(input_dims), input_data, in StridedSlice()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 367 HANDLE_OPERATORTYPENAME_CASE(StridedSlice) in OperatorTypeName()
|