Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dsparse_expander_common.h56 StatusOr<mlir::Value> GetDenseShapesFromSparseTensor(mlir::Value value);
H A Dsparse_expander_common.cc71 StatusOr<mlir::Value> GetDenseShapesFromSparseTensor(mlir::Value value) { in GetDenseShapesFromSparseTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/
H A Dmatmul_sparse_expander.cc42 GetDenseShapesFromSparseTensor(op->getOperand(0)).ValueOrDie(), in ExpandOp()