Searched defs:tensor_a (Results 1 – 7 of 7) sorted by relevance
98 TfLiteTensor tensor_a, tensor_b; in TEST() local139 TfLiteTensor tensor_a, tensor_b; in TEST() local
42 auto tensor_a = in TEST() local
44 const auto& tensor_a = context->tensors[tensor_a_index]; in PopulateSubGraph() local
199 def assert_close_tensors(tensor_a, tensor_b): argument
143 bool isNonZeroDimEqual(Tensor& tensor_a, Tensor& tensor_b) { in isNonZeroDimEqual()
621 const auto tensor_a = weight; in _sparse_semi_structured_linear() local
4509 ITensorProxyPtr tensor_a = nullptr; in ConvertFullyConnectedImpl() local4624 ITensorProxyPtr tensor_a = convert_to_itensor(input_a); in ConvertMatMulImpl() local