Searched defs:sparse_values (Results 1 – 15 of 15) sorted by relevance
67 xla::XlaOp sparse_values = context->Input(2); in Compile() local
106 def f(sparse_values, default_value): argument
967 sparse_values, argument
74 const Tensor& sparse_values = c->input(2); in Compute() local
219 Tensor sparse_values(DT_FLOAT, TensorShape({N})); in BM_SparseToDense() local
148 OpOutputList sparse_values; in Compute() local244 OpOutputList sparse_values; in Compute() local
175 std::vector<NodeBuilder::NodeOut> sparse_values; in GetGraphs() local
32 sparse_values, argument
106 const Tensor& sparse_values, in set_input_tensors()
88 std::vector<Tensor> sparse_values; member
381 sparse_values, argument
422 const auto sparse_values = in Compute() local807 const auto sparse_values = in Compute() local
743 def _do_batch_all_reduce_sparse(self, reduce_op, sparse_values): argument
556 const Tensor& sparse_values) { in GetSparseFunctor()