Searched refs:tensorAt (Results 1 – 3 of 3) sorted by relevance
219 auto& arg = spec.tensorAt(tensor_arg_spec_offset++); in specializeTypes()225 auto& arg = spec.tensorAt(tensor_arg_spec_offset++); in specializeTypes()
147 const ArgumentInfo& tensorAt(size_t i) const { in tensorAt() function414 out << spec.tensorAt(i);
180 ASSERT_TRUE(isEqual(a.tensorAt(i), list[i].toTensor())); in TEST()