Searched refs:_assert_scalar (Results 1 – 15 of 15) sorted by relevance
| /external/pytorch/torch/fx/passes/ |
| D | runtime_assert.py | 249 torch.ops.aten._assert_scalar.default,
321 torch.ops.aten._assert_scalar.default,
576 torch.ops.aten._assert_scalar.default,
586 torch.ops.aten._assert_scalar.default,
|
| D | tools_common.py | 270 torch.ops.aten._assert_scalar.default,
|
| /external/executorch/exir/passes/ |
| D | remove_graph_asserts_pass.py | 29 torch.ops.aten._assert_scalar.default,
|
| D | replace_aten_with_edge_pass.py | 18 torch.ops.aten._assert_scalar.default,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 447 serialized = _assert_scalar(serialized, "serialized")
837 serialized = _assert_scalar(serialized, "serialized")
1132 def _assert_scalar(value, name): function
|
| /external/pytorch/torch/onnx/_internal/exporter/ |
| D | _fx_passes.py | 44 torch.ops.aten._assert_scalar.default,
|
| /external/executorch/extension/llm/modules/test/ |
| D | test_position_embeddings.py | 100 torch.ops.aten._assert_scalar.default,
204 torch.ops.aten._assert_scalar.default,
|
| /external/pytorch/torch/_export/passes/ |
| D | add_runtime_assertions_for_constraints_pass.py | 177 if node.target != torch.ops.aten._assert_scalar.default:
|
| /external/executorch/exir/verification/ |
| D | verifier.py | 238 torch.ops.aten._assert_scalar.default,
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorCompare.cpp | 429 void _assert_scalar(const Scalar& scalar, c10::string_view assert_msg) { in _assert_scalar() function
434 _assert_scalar(scalar, assert_msg); in _functional_assert_scalar()
|
| /external/pytorch/docs/source/ |
| D | torch.compiler_cudagraph_trees.rst | 252 aten._assert_scalar
268 aten._assert_scalar
|
| /external/pytorch/torch/fx/ |
| D | node.py | 50 _ops.aten._assert_scalar.default,
|
| /external/pytorch/test/export/ |
| D | test_passes.py | 665 ep.graph, torch.ops.aten._assert_scalar.default
|
| D | test_export.py | 6721 node.target == torch.ops.aten._assert_scalar.default
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 2715 @register_lowering(aten._assert_scalar)
2716 def _assert_scalar(data, msg): function
|