Searched defs:tensor_b (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
44 auto tensor_b = in TEST() local
45 const auto& tensor_b = context->tensors[tensor_b_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()
622 const auto tensor_b = in _sparse_semi_structured_linear() local
4625 ITensorProxyPtr tensor_b = convert_to_itensor(input_b); in ConvertMatMulImpl() local