Searched refs:get_tensor_value (Results 1 – 8 of 8) sorted by relevance
11 get_tensor_value,58 input_shape = get_tensor_value(xnn_graph.xvalues[input_id]).dims145 input_shape = get_tensor_value(xnn_graph.xvalues[input_id]).dims
12 get_tensor_value,67 input_shape = get_tensor_value(xnn_graph.xvalues[input_id]).dims
11 get_tensor_value,106 get_tensor_value(xnn_graph.xvalues[input_id]).dims
12 get_tensor_value,46 input_shape = get_tensor_value(xnn_graph.xvalues[kwargs["input_id"]]).dims
12 get_tensor_value,79 new_shape = get_tensor_value(xnn_graph.xvalues[output_id]).dims
69 def get_tensor_value(xvalue: XValue) -> XNNTensorValue: function
632 … self.assertEqual(len(observer_dict['fc1.module.activation_post_process'].get_tensor_value()),634 … self.assertEqual(observer_dict['fc1.module.activation_post_process'].get_tensor_value()[0],645 self.assertTrue(torch.equal(obs.get_tensor_value()[0], scripted.get_tensor_value()[0]))650 self.assertTrue(torch.equal(obs.get_tensor_value()[0], loaded.get_tensor_value()[0]))
1524 def get_tensor_value(self): member in RecordingObserver