Home
last modified time | relevance | path

Searched refs:kStridesTensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DStridedSlice.cpp89 context->getInputBuffer<int32_t>(kStridesTensor), in executeTyped()
105 const Shape& stridesShape = context->getInputShape(kStridesTensor); in prepare()
121 const int32_t* stridesData = context->getInputBuffer<int32_t>(kStridesTensor); in prepare()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dstrided_slice.cc47 constexpr int kStridesTensor = 3; variable
56 strides = GetInput(context, node, kStridesTensor); in StridedSliceContext()
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DStridedSlice.h28 constexpr uint32_t kStridesTensor = 3; variable