Home
last modified time | relevance | path

Searched refs:get_tensor_value (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/executorch/backends/xnnpack/operators/
H A Dop_max_dim.py11 get_tensor_value,
58 input_shape = get_tensor_value(xnn_graph.xvalues[input_id]).dims
145 input_shape = get_tensor_value(xnn_graph.xvalues[input_id]).dims
H A Dop_mean_dim.py12 get_tensor_value,
67 input_shape = get_tensor_value(xnn_graph.xvalues[input_id]).dims
H A Dop_static_constant_pad.py11 get_tensor_value,
106 get_tensor_value(xnn_graph.xvalues[input_id]).dims
H A Dop_max_pool2d.py12 get_tensor_value,
46 input_shape = get_tensor_value(xnn_graph.xvalues[kwargs["input_id"]]).dims
H A Dop_to_copy.py12 get_tensor_value,
79 new_shape = get_tensor_value(xnn_graph.xvalues[output_id]).dims
H A Dnode_visitor.py69 def get_tensor_value(xvalue: XValue) -> XNNTensorValue: function
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_workflow_module.py632 … 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]))
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dobserver.py1524 def get_tensor_value(self): member in RecordingObserver