Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSerialStackImpl.h71 const Tensor& first_tensor = tensors[0]; in can_use_native_serial_stack_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter_test.cc1123 TfLiteTensor* first_tensor = context->tensors; in TEST() local
1143 TfLiteTensor* first_tensor = context->tensors; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
H A Dtfr_ops.cc212 int first_tensor = -1, last_tensor = -1, first_tensor_list = -1, in verify() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dops.cpp849 const auto& first_tensor = inputs[0]; in inputsAreScalars() local