Home
last modified time | relevance | path

Searched refs:ops_one_tensor_in_shape_transform (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/
H A Dop_registry.h29 std::shared_ptr<OperatorSet> ops_one_tensor_in_shape_transform();
H A Dop_registry.cpp69 std::shared_ptr<OperatorSet> ops_one_tensor_in_shape_transform() { in ops_one_tensor_in_shape_transform() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Ddtype_analysis.cpp319 *ops_one_tensor_in_shape_transform(), setIfAllDtypeMatch); in buildDtypeRuleRegistry()