Searched refs:SliceAttributes (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | strided_slice.h | 28 StridedSlice(const OperationDef& definition, const SliceAttributes& attr);
41 SliceAttributes attributes_;
45 const SliceAttributes& attr);
|
| D | strided_slice.cc | 26 bool Is4Aligned(const SliceAttributes& attr) { in Is4Aligned()
30 int4 GetOffset(const SliceAttributes& attr, int src_width, int src_height, in GetOffset()
78 const SliceAttributes& attr) in StridedSlice()
177 const SliceAttributes& attr) { in CreateStridedSlice()
|
| D | strided_slice_test_util.cc | 38 SliceAttributes attr; in StridedSliceTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | slice_test.cc | 44 SliceAttributes attr; in TEST()
67 SliceAttributes attr; in TEST()
90 SliceAttributes attr; in TEST()
113 SliceAttributes attr; in TEST()
136 SliceAttributes attr; in TEST()
159 SliceAttributes attr; in TEST()
|
| D | slice.cc | 38 const auto& attr = absl::any_cast<const SliceAttributes&>(ctx.op_attr); in GenerateCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 469 struct SliceAttributes { struct
480 BHWC CalculateOutputShape(const BHWC& input, const SliceAttributes& attr); argument
|
| D | model_builder.cc | 1638 SliceAttributes attr; in Parse()
1726 SliceAttributes* attr) { in UpdateIfNegative()
1904 SliceAttributes attr; in Parse()
1945 SliceAttributes* attr) { in UpdateWithMask()
1975 SliceAttributes* attr) { in UpdateIfNegative()
2008 SliceAttributes* attr) { in ReadAttribsWithBatch()
2026 const BHWC& input_shape, SliceAttributes* attr) { in ReadAttribsWithoutBatch()
|
| D | operations.cc | 549 BHWC CalculateOutputShape(const BHWC& input, const SliceAttributes& attr) { in CalculateOutputShape()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | simple_selectors.h | 69 void SelectStridedSlice(const SliceAttributes& attr, const OperationDef& op_def,
|
| D | simple_selectors.cc | 155 void SelectStridedSlice(const SliceAttributes& attr, const OperationDef& op_def, in SelectStridedSlice()
|
| D | operation_selector.cc | 480 auto attr = absl::any_cast<SliceAttributes>(node.operation.attributes); in GPUOperationFromNode()
|