Searched defs:dense_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | sparse_tensor.py | 106 def __init__(self, indices, values, dense_shape): argument
181 def dense_shape(self): member in SparseTensor
|
| D | ops.py | 1358 def __init__(self, values, indices, dense_shape=None): argument
1376 def dense_shape(self): member in IndexedSlices
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | sparse_tensor_slice_dataset_op.cc | 65 std::vector<int64> dense_shape; in AsGraphDefInternal() local
222 const Tensor* dense_shape; in MakeDataset() local
|
| D | dense_to_sparse_batch_dataset_op.cc | 158 Tensor dense_shape(ctx->allocator({}), DT_INT64, {row_ndims + 1}); in GetNextInternal() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.h | 181 TensorShape dense_shape; in Init() local
247 TensorShape dense_shape; in Init() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | embedding_lookup_sparse.cc | 144 TfLiteTensor* dense_shape = GetInput(context, node, 2); in Eval() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op.cc | 73 const auto dense_shape = dense_shape_t->vec<int64>(); in Compute() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 589 dense_shape = np.array([2, 3]).astype(np.int64) variable in SparseReduceTest
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 418 const auto dense_shape = in Compute() local
|