Home
last modified time | relevance | path

Searched refs:GetValuesFromSparseTensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dsparse_expander_common.h52 StatusOr<mlir::Value> GetValuesFromSparseTensor(mlir::Value value);
H A Dsparse_expander_common.cc62 StatusOr<mlir::Value> GetValuesFromSparseTensor(mlir::Value value) { in GetValuesFromSparseTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/
H A Dmatmul_sparse_expander.cc41 GetValuesFromSparseTensor(op->getOperand(0)).ValueOrDie(), in ExpandOp()
H A Ddynamic_enqueue_sparse_expander.cc91 GetValuesFromSparseTensor(sparse_feature_value).ValueOrDie()); in ExpandOp()