Home
last modified time | relevance | path

Searched defs:StridedSlice (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.h31 struct StridedSlice { struct
32 void operator()(const Device& d, typename TTypes<T, NDIMS>::Tensor output, in operator()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dstrided_slice.h30 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
97 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
108 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
Dlegacy_reference_ops.h936 inline void StridedSlice(const T* input_data, const Dims<4>& input_dims, in StridedSlice() function
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dstrided_slice.cc77 StridedSlice::StridedSlice(const OperationDef& definition, in StridedSlice() function in tflite::gpu::StridedSlice
84 StridedSlice::StridedSlice(StridedSlice&& operation) in StridedSlice() function in tflite::gpu::StridedSlice
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dstrided_slice_test.cc30 TEST_F(OpenCLOperationTest, StridedSlice) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_strided_slice.cc28 void StridedSlice(StridedSliceOperator const& op, Array const& input_array, in StridedSlice() function
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc480 TEST_F(OperatorTest, StridedSlice) { in TEST_F() argument
Doperator.cc1217 class StridedSlice class
/external/tensorflow/tensorflow/compiler/xla/service/
Dsharding_propagation_test.cc1284 TEST_P(ParameterizedMetadataTest, StridedSlice) { in TEST_P() argument
Dhlo_evaluator_test.cc2968 TEST_P(HloEvaluatorBf16Test, StridedSlice) { in TEST_P() argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5441 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
5519 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
5530 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
Dlegacy_optimized_ops.h4952 inline void StridedSlice(const T* input_data, const Dims<4>& input_dims, in StridedSlice() function
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3203 TEST_F(OpTest, StridedSlice) { in TEST_F() argument
/external/tensorflow/tensorflow/go/op/
Dwrappers.go33858 func StridedSlice(scope *Scope, input tf.Output, begin tf.Output, end tf.Output, strides tf.Output,… func