Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dsparse_expander_common.h48 StatusOr<mlir::Value> GetIndicesFromSparseTensor(mlir::Value value);
H A Dsparse_expander_common.cc53 StatusOr<mlir::Value> GetIndicesFromSparseTensor(mlir::Value value) { in GetIndicesFromSparseTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/
H A Dmatmul_sparse_expander.cc40 GetIndicesFromSparseTensor(op->getOperand(0)).ValueOrDie(), in ExpandOp()
H A Ddynamic_enqueue_sparse_expander.cc88 GetIndicesFromSparseTensor(sparse_feature_value).ValueOrDie())); in ExpandOp()