Searched refs:SliceAttributes (Results 1 – 14 of 14) sorted by relevance
| /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 | 41 const auto& attr = std::any_cast<const SliceAttributes&>(ctx.op_attr); in GenerateCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | strided_slice.h | 30 StridedSlice(const OperationDef& definition, const SliceAttributes& attr);
43 SliceAttributes attributes_;
47 const SliceAttributes& attr);
|
| D | strided_slice.cc | 27 bool Is4Aligned(const SliceAttributes& attr) { in Is4Aligned()
31 int4 GetOffset(const SliceAttributes& attr, int src_width, int src_height, in GetOffset()
79 const SliceAttributes& attr) in StridedSlice()
178 const SliceAttributes& attr) { in CreateStridedSlice()
|
| D | strided_slice_test_util.cc | 39 SliceAttributes attr; in StridedSliceTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | conv_pointwise.cc | 136 absl::any_cast<SliceAttributes>(node_context->node->operation.attributes); in IsSliceNode()
169 absl::any_cast<SliceAttributes>(slice_node.node->operation.attributes); in GetOffset()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 511 struct SliceAttributes { struct
522 BHWC CalculateOutputShape(const BHWC& input, const SliceAttributes& attr); argument
|
| D | model_builder.cc | 2085 SliceAttributes attr; in Parse()
2173 SliceAttributes* attr) { in UpdateIfNegative()
2369 SliceAttributes attr; in Parse()
2410 SliceAttributes* attr) { in UpdateWithMask()
2440 SliceAttributes* attr) { in UpdateIfNegative()
2473 SliceAttributes* attr) { in ReadAttribsWithBatch()
2491 const BHWC& input_shape, SliceAttributes* attr) { in ReadAttribsWithoutBatch()
|
| D | operations.cc | 588 BHWC CalculateOutputShape(const BHWC& input, const SliceAttributes& attr) { in CalculateOutputShape()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | simple_selectors.h | 78 void SelectStridedSlice(const SliceAttributes& attr, const OperationDef& op_def,
|
| D | simple_selectors.cc | 187 void SelectStridedSlice(const SliceAttributes& attr, const OperationDef& op_def, in SelectStridedSlice()
|
| D | operation_selector.cc | 650 auto attr = absl::any_cast<SliceAttributes>(node.operation.attributes); in GPUOperationFromNodePart0()
686 SliceAttributes new_attr; in GPUOperationFromNodePart0()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | remove_noop_test.cc | 261 SliceAttributes attr; in TEST()
319 SliceAttributes attr; in TEST()
|
| D | remove_noop.cc | 155 absl::any_cast<const SliceAttributes&>(node->operation.attributes); in ApplyToNode()
|