Home
last modified time | relevance | path

Searched refs:Tensors (Results 1 – 25 of 94) sorted by relevance

1234

/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DTensorTest.java54 try (Tensor<String> t = Tensors.create(strings)) { in createWithByteBuffer()
175 try (Tensor<Integer> tints = Tensors.create(ints); in writeTo()
176 Tensor<Float> tfloats = Tensors.create(floats); in writeTo()
177 Tensor<Double> tdoubles = Tensors.create(doubles); in writeTo()
178 Tensor<Long> tlongs = Tensors.create(longs); in writeTo()
179 Tensor<Boolean> tbools = Tensors.create(bools)) { in writeTo()
302 try (Tensor<Float> t = Tensors.create(2.718f)) { in scalars()
309 try (Tensor<Double> t = Tensors.create(3.1415)) { in scalars()
316 try (Tensor<Integer> t = Tensors.create(-33)) { in scalars()
323 try (Tensor<Long> t = Tensors.create(8589934592L)) { in scalars()
[all …]
DGraphOperationBuilderTest.java51 Tensor<Integer> t = Tensors.create(1)) { in failOnUseAfterBuild()
67 Tensor<Integer> t = Tensors.create(1)) { in failOnUseAfterGraphClose()
88 try (Tensor<Integer> t = Tensors.create(1)) { in setAttr()
169 Tensor<Boolean> yes = Tensors.create(true); in addControlInput()
170 Tensor<Boolean> no = Tensors.create(false)) { in addControlInput()
DSessionTest.java36 try (Tensor<Integer> x = Tensors.create(new int[][] {{5}, {7}}); in runUsingOperationNames()
53 try (Tensor<Integer> x = Tensors.create(new int[][] {{5}, {7}}); in runUsingOperationHandles()
85 try (Tensor<Integer> fed = Tensors.create(new int[] {4, 3, 2, 1}); in runUsingColonSeparatedNames()
105 try (Tensor<Integer> x = Tensors.create(new int[][] {{5}, {7}})) { in runWithMetadata()
DGraphTest.java155 try (Tensor<Float> c1 = Tensors.create(3.0f); in addGradientsToGraph()
156 Tensor<Float> c2 = Tensors.create(2.0f); in addGradientsToGraph()
188 try (Tensor<Float> c = Tensors.create(3.0f); in addGradientSumsToGraph()
219 try (Tensor<Float> c = Tensors.create(3.0f); in addGradientsWithInitialValuesToGraph()
298 try (Tensor<Integer> c = Tensors.create(2); in buildWhileLoopSingleInput()
348 try (Tensor<Integer> c1 = Tensors.create(2); in buildWhileLoopMultipleInputs()
349 Tensor<Integer> c2 = Tensors.create(5); in buildWhileLoopMultipleInputs()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Batch.pbtxt6 container/shared_name in the same device, some will output zero-shaped Tensors
7 and others will output Tensors of size up to max_batch_size.
9 All Tensors in in_tensors are batched together (so, for example, labels and
17 start, and length of elements of each set of Tensors present in batched_tensors.
33 batched_tensors: Either empty tensors or a batch of concatenated Tensors.
Dapi_def_FusedBatchNorm.pbtxt72 The data type for the elements of input and output Tensors.
96 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
97 The size of 1D Tensors matches the dimension C of the 4D Tensors.
Dapi_def_ParseSingleSequenceExample.pbtxt21 A list of Ncontext_sparse string Tensors (scalars).
29 A list of Ncontext_dense string Tensors (scalars).
37 A list of Nfeature_list_sparse string Tensors
45 A list of Nfeature_list_dense string Tensors (scalars).
53 A list of Ncontext_dense Tensors (some may be empty).
Dapi_def_ParseSequenceExample.pbtxt22 A list of Ncontext_dense Tensors (some may be empty).
44 A list of Ncontext_sparse string Tensors (scalars).
52 A list of Ncontext_dense string Tensors (scalars).
60 A list of Nfeature_list_sparse string Tensors
68 A list of Nfeature_list_dense string Tensors (scalars).
Dapi_def_FusedBatchNormV3.pbtxt79 The data type for the elements of input and output Tensors.
109 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
110 The size of 1D Tensors matches the dimension C of the 4D Tensors.
Dapi_def_FusedBatchNormGrad.pbtxt74 The data type for the elements of input and output Tensors.
99 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
100 The size of 1D Tensors matches the dimension C of the 4D Tensors.
Dapi_def_FusedBatchNormV2.pbtxt72 The data type for the elements of input and output Tensors.
102 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
103 The size of 1D Tensors matches the dimension C of the 4D Tensors.
Dapi_def_FusedBatchNormGradV2.pbtxt74 The data type for the elements of input and output Tensors.
105 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
106 The size of 1D Tensors matches the dimension C of the 4D Tensors.
Dapi_def_BoostedTreesSparseAggregateStats.pbtxt25 int32; Rank 2 indices of feature sparse Tensors (shape=[number of sparse entries, 2]).
34 int32; Rank 1 values of feature sparse Tensors (shape=[number of sparse entries]).
42 int32; Rank 1 dense shape of feature sparse Tensors (shape=[2]).
49 int32; Rank 2 indices of summary sparse Tensors (shape=[number of non zero statistics, 4])
Dapi_def_FusedBatchNormGradV3.pbtxt82 The data type for the elements of input and output Tensors.
113 Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
114 The size of 1D Tensors matches the dimension C of the 4D Tensors.
Dapi_def_BoostedTreesBucketize.pbtxt13 float; List of Rank 1 Tensors each containing the bucket boundaries for a single
20 int; List of Rank 1 Tensors each containing the bucketized values for a single feature.
Dapi_def_BoostedTreesMakeQuantileSummaries.pbtxt7 float; List of Rank 1 Tensors each containing values for a single feature.
25 float; List of Rank 2 Tensors each containing the quantile summary
Dapi_def_EnqueueTPUEmbeddingSparseBatch.pbtxt7 A list of rank 1 Tensors specifying the training example and
17 A list of rank 1 Tensors, indices into the embedding tables.
23 A list of rank 1 Tensors containing per sample -- i.e. per
Dapi_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt7 A list of rank 1 Tensors specifying the training example to
15 A list of rank 1 Tensors, indices into the embedding tables.
22 A list of rank 1 Tensors containing per training example
Dapi_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt7 A list of rank 1 Tensors specifying the break points for splitting
16 A list of rank 1 Tensors, indices into the embedding tables.
23 A list of rank 1 Tensors containing per training example
Dapi_def_RaggedTensorToVariant.pbtxt7 A list of one or more Tensors representing the splits of the input
41 splits and values Tensors of the `RaggedTensor`. Then the 1-D `variant` Tensor
Dapi_def_ParseExample.pbtxt23 A list of Nsparse string Tensors (scalars).
30 A list of Ndense string Tensors (scalars).
37 A list of Ndense Tensors (some may be empty).
/external/tensorflow/tensorflow/core/kernels/
Dtensor_list.h67 TensorList() : tensors_(new Tensors) {} in TensorList()
145 class Tensors : public core::RefCounted {
149 Tensors* tensors_;
Dtensor_map.h69 TensorMap() : tensors_(new Tensors) {} in TensorMap()
166 class Tensors : public core::RefCounted {
170 Tensors* tensors_;
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/op/core/
DGradientsTest.java30 import org.tensorflow.Tensors;
53 try (Tensor<Float> c = Tensors.create(3.0f); in createGradients()
80 try (Tensor<Float> c = Tensors.create(3.0f); in createGradientsWithSum()
106 try (Tensor<Float> c = Tensors.create(3.0f); in createGradientsWithInitialValues()
/external/tensorflow/tensorflow/java/src/gen/resources/
DTensors.java.tmpl10 public final class Tensors {
11 private Tensors() {}

1234