Home
last modified time | relevance | path

Searched defs:end_vec (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DLerpKernel.cpp74 [=](bVec self_vec, bVec end_vec) -> bVec { in lerp_scalar_kernel()
91 [=](hVec self_vec, hVec end_vec) -> hVec { in lerp_scalar_kernel()
122 [=](bVec self_vec, bVec end_vec, bVec weight_vec) -> bVec { in lerp_tensor_kernel()
137 [=](hVec self_vec, hVec end_vec, hVec weight_vec) -> hVec { in lerp_tensor_kernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drun_handler_util.cc82 std::vector<std::uint_fast32_t>* end_vec) { in ComputeInterOpSchedulingRanges()
115 std::vector<std::uint_fast32_t>* end_vec) { in ComputeInterOpStealingRanges()
H A Drun_handler_util_test.cc98 std::vector<std::uint_fast32_t> end_vec(num_inter_op_threads); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc2595 auto end_vec = end_tensor.flat<int32>(); in ConvertSlice() local