Home
last modified time | relevance | path

Searched refs:tensor_or_scalar (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder.cc116 TensorOrScalar* tensor_or_scalar) { in ParseInputsWithConstTensor() argument
154 *tensor_or_scalar = tensor.data[0]; in ParseInputsWithConstTensor()
159 *tensor_or_scalar = std::move(tensor); in ParseInputsWithConstTensor()
167 *tensor_or_scalar = std::move(tensor); in ParseInputsWithConstTensor()
171 *tensor_or_scalar = std::move(tensor); in ParseInputsWithConstTensor()
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py12365 for tensor_or_scalar in rhs:
12366 funcs_str = funcs_template.format(a_construct=tensor, b_construct=tensor_or_scalar)