Home
last modified time | relevance | path

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

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