Home
last modified time | relevance | path

Searched defs:sparse_indices (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dsparse_column_iterable.h115 const TTypes<int64>::ConstMatrix& sparse_indices() const { return ix_; } in sparse_indices() function
/external/tensorflow/tensorflow/compiler/tests/
Dsparse_to_dense_op_test.py30 def _SparseToDense(sparse_indices, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_to_dense_op_py_test.py30 def _SparseToDense(sparse_indices, argument
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dinput_data.cc105 const Tensor& sparse_indices, in set_input_tensors()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.h87 std::vector<Tensor> sparse_indices; member
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_sparse_kernel.cc69 auto sparse_indices = sparse_indices_out->tensor<int64, 2>(); in Compute() local
Dsparse_to_dense_op_test.cc218 Tensor sparse_indices(DT_INT32, TensorShape({N, NDIM})); in BM_SparseToDense() local
Dexample_parsing_ops.cc147 OpOutputList sparse_indices; in Compute() local
243 OpOutputList sparse_indices; in Compute() local
Dsdca_ops_test.cc160 std::vector<NodeBuilder::NodeOut> sparse_indices; in GetGraphs() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h353 SparseIndexArray* sparse_indices() const { return sparse_indices_; } in sparse_indices() function
354 void set_sparse_indices(SparseIndexArray* sparse_indices) { in set_sparse_indices()
Dxla_data.proto379 repeated int64 sparse_indices = 14; field
Dliteral.cc198 const SparseIndexArray* LiteralBase::sparse_indices( in sparse_indices() function in xla::LiteralBase
203 SparseIndexArray* MutableLiteralBase::sparse_indices( in sparse_indices() function in xla::MutableLiteralBase
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc424 const auto sparse_indices = in Compute() local
809 const auto sparse_indices = in Compute() local
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py965 def sparse_to_dense(sparse_indices, argument
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.cc555 GetFeatureFnType GetSparseFunctor(const Tensor& sparse_indices, in GetSparseFunctor()