Home
last modified time | relevance | path

Searched defs:slice_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcuda_caching_host_allocator_test.cpp43 auto slice_tensor = in TEST() local
H A Dxpu_caching_host_allocator_test.cpp45 auto slice_tensor = in TEST() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_glu.cpp81 void slice_tensor( in slice_tensor() function
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DSlice.cpp283 void slice_tensor(ComputeGraph& graph, const std::vector<ValueRef>& args) { in slice_tensor() function
/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dtensor_bundle.cc481 const Tensor& slice_tensor) { in AddSlice()
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dops.py1389 def slice_tensor(func, *args, **kwargs): function