Home
last modified time | relevance | path

Searched defs:dense_shape (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor.py109 def __init__(self, indices, values, dense_shape): argument
178 def dense_shape(self): member in SparseTensor
Dops.py1653 def __init__(self, values, indices, dense_shape=None): argument
1671 def dense_shape(self): member in IndexedSlices
/external/tensorflow/tensorflow/core/kernels/data/
Dsparse_tensor_slice_dataset_op.cc68 std::vector<int64> dense_shape; in AsGraphDefInternal() local
230 const Tensor* dense_shape; in MakeDataset() local
Dsparse_tensor_slice_dataset_op_test.cc65 Tensor dense_shape; member
186 Tensor dense_shape = CreateTensor<int64>(TensorShape({NDIM}), {5, 5}); in TEST_F() local
306 Tensor dense_shape = CreateTensor<int64>(TensorShape({NDIM}), {5, 5}); in TEST_F() local
414 Tensor dense_shape = CreateTensor<int64>(TensorShape({NDIM}), {5, 5}); in TEST_F() local
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.h182 TensorShape dense_shape; in Init() local
248 TensorShape dense_shape; in Init() local
Dexample_proto_fast_parsing.cc2003 TensorShape dense_shape, example_shape; in FastParseSequenceExample() local
2198 TensorShape dense_shape, row_shape; in FastParseSequenceExample() local
/external/tensorflow/tensorflow/contrib/distribute/python/
Dcross_device_ops_test.py83 def _make_indexed_slices(values, indices, dense_shape, device): argument
92 def _make_mirrored_indexed_slices(devices, values, indices, dense_shape): argument
/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse.cc144 const TfLiteTensor* dense_shape = GetInput(context, node, 2); in Eval() local
/external/tensorflow/tensorflow/core/kernels/
Dsparse_fill_empty_rows_op.cc73 const auto dense_shape = dense_shape_t->vec<int64>(); in Compute() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc171 Tensor dense_shape(ctx->allocator({}), DT_INT64, {row_ndims + 1}); in GetNextInternal() local
Dparse_example_dataset_op.cc58 TensorShape dense_shape; in ParseExampleDatasetOp() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dunicode_decode_op_test.py87 def _make_sparse_tensor(indices, values, dense_shape, dtype=np.int32): argument
Dsparse_ops_test.py612 dense_shape = np.array([2, 3]).astype(np.int64) variable in SparseReduceTest
/external/tensorflow/tensorflow/python/data/util/
Dstructure.py504 def __init__(self, dtype, dense_shape): argument
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc426 const auto dense_shape = in Compute() local
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc3026 tensorflow::Safe_PyObjectPtr dense_shape( local
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py3829 def generate_index(dense_shape): argument