Home
last modified time | relevance | path

Searched defs:dim_int (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform_patterns.cc341 IntegerAttr dim_int = ExtractSingleElementAsInteger(split_dim_attr); in matchAndRewrite() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReduceOps.cpp282 TORCH_META_FUNC2(prod, dim_int) in TORCH_META_FUNC2() argument