Home
last modified time | relevance | path

Searched defs:split_dim_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc326 mlir::ElementsAttr split_dim_attr; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3321 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local
3384 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local
3500 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1529 DenseIntElementsAttr split_dim_attr; in VerifySplitInputAndSplitDim() local