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.cc322 mlir::ElementsAttr split_dim_attr; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3524 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local
3587 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local
3692 DenseIntElementsAttr split_dim_attr; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1473 DenseIntElementsAttr split_dim_attr; in VerifySplitInputAndSplitDim() local