Searched refs:_get_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | utils_impl.py | 168 def _get_tensor(name): function
173 return _get_tensor(tensor_info.name)
176 _get_tensor(tensor_info.coo_sparse.indices_tensor_name),
177 _get_tensor(tensor_info.coo_sparse.values_tensor_name),
178 _get_tensor(tensor_info.coo_sparse.dense_shape_tensor_name))
183 components = [_get_tensor(component.name) for component in
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | micro_pipeline_tp.py | 58 def _get_tensor(node: torch.fx.Node) -> torch.Tensor: function
336 args=(new_node, list(_get_tensor(mm_node).shape)),
420 *_get_tensor(matmul_input_node).shape[:-1],
421 _get_tensor(B_node).shape[-1],
424 if _get_tensor(reshape_node).shape != matmul_out_shape:
544 if gather_dim >= len(_get_tensor(shard_node).shape) - 1:
567 _get_tensor(shard_node),
722 _get_tensor(matmul.A_node),
|
| /external/executorch/backends/xnnpack/operators/ |
| D | quant_params.py | 187 def _get_tensor(node): function
192 scale = _get_tensor(scale)
193 zp = _get_tensor(zp)
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_test.py | 104 def _get_tensor(self, is_finite): member in DynamicLossScaleTest
134 grad = self._get_tensor(is_finite)
|
| /external/executorch/backends/qualcomm/builders/ |
| D | node_visitor.py | 97 def _get_tensor(node, index): function
108 tensor = _get_tensor(input_node, idx)
|