Home
last modified time | relevance | path

Searched refs:minibatch (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AddManySparseToTensorsMap.pbtxt6 2-D. The `indices` of the minibatch `SparseTensor`.
13 1-D. The `values` of the minibatch `SparseTensor`.
19 1-D. The `shape` of the minibatch `SparseTensor`.
20 The minibatch size `N == sparse_shape[0]`.
43 summary: "Add an `N`-minibatch `SparseTensor` to a `SparseTensorsMap`, return `N` handles."
50 An `N`-minibatch of `SparseTensor` objects is represented as a `SparseTensor`
52 the minibatch size `N == sparse_shape[0]`.
55 dimension is treated as the minibatch dimension. Elements of the `SparseTensor`
60 The `SparseTensor` values can then be read out as part of a minibatch by passing
Dapi_def_SerializeManySparse.pbtxt6 2-D. The `indices` of the minibatch `SparseTensor`.
12 1-D. The `values` of the minibatch `SparseTensor`.
18 1-D. The `shape` of the minibatch `SparseTensor`.
28 summary: "Serialize an `N`-minibatch `SparseTensor` into an `[N, 3]` `Tensor` object."
31 is treated as the minibatch dimension. Elements of the `SparseTensor`
36 The minibatch size `N` is extracted from `sparse_shape[0]`.
Dapi_def_TakeManySparseFromTensorsMap.pbtxt13 2-D. The `indices` of the minibatch `SparseTensor`.
19 1-D. The `values` of the minibatch `SparseTensor`.
25 1-D. The `shape` of the minibatch `SparseTensor`.
52 `N` is the minibatch size and the rows correspond to the output handles of
61 for the corresponding dimensions. Its first shape value is `N`, the minibatch
Dapi_def_DeserializeManySparse.pbtxt16 summary: "Deserialize and concatenate `SparseTensors` from a serialized minibatch."
19 `N` is the minibatch size and the rows correspond to packed outputs of
27 for the corresponding dimensions. Its first shape value is `N`, the minibatch
Dapi_def_ParseExample.pbtxt71 of elements of length D1 * .... * DN, across all minibatch entries
72 in the input. Any minibatch entry with less than M blocks of elements of
Dapi_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt13 This is the label for the given minibatch entry.
Dapi_def_ParseExampleV2.pbtxt102 of elements of length D1 * .... * DN, across all minibatch entries
103 in the input. Any minibatch entry with less than M blocks of elements of
Dapi_def_DecodeProtoV2.pbtxt77 the largest number of repeated elements seen in the input minibatch. (The
79 repeat counts for each example in the minibatch can be found in the `sizes`
Dapi_def_AddSparseToTensorsMap.pbtxt50 The `SparseTensor` can then be read out as part of a minibatch by passing
Dapi_def_PaddingFIFOQueue.pbtxt54 size of any given element in the minibatch. See below for details.
Dapi_def_PaddingFIFOQueueV2.pbtxt56 size of any given element in the minibatch. See below for details.
Dapi_def_ExperimentalParseExampleDataset.pbtxt52 minibatch elements smaller than the maximum number of blocks for the
Dapi_def_ParseExampleDataset.pbtxt52 minibatch elements smaller than the maximum number of blocks for the
Dapi_def_ParseExampleDatasetV2.pbtxt52 minibatch elements smaller than the maximum number of blocks for the
/external/tensorflow/tensorflow/core/kernels/
Dserialize_sparse_op.cc149 Status operator()(sparse::GroupIterable* minibatch, in operator ()()
172 for (const auto& subset : *minibatch) { in operator ()()
251 Status operator()(sparse::GroupIterable* minibatch, in operator ()()
271 for (const auto& subset : *minibatch) { in operator ()()
389 sparse::GroupIterable minibatch = input_st.group({0}); in Compute() local
391 OP_REQUIRES_OK(context, SerializeGroups<T, U>()(&minibatch, output_shape, N, in Compute()
Dsparse_tensors_map_ops.cc282 sparse::GroupIterable minibatch = input_st.group({0}); in Compute() local
283 for (const auto& subset : minibatch) { in Compute()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1195 auto first_example_of_minibatch = [&](size_t minibatch) -> size_t { in FastParseExample() argument
1196 return (serialized.size() * minibatch) / num_minibatches; in FastParseExample()
1212 auto ProcessMiniBatch = [&](size_t minibatch) { in FastParseExample() argument
1213 sparse_buffers[minibatch].resize(config.sparse.size()); in FastParseExample()
1214 varlen_dense_buffers[minibatch].resize(config.dense.size()); in FastParseExample()
1215 ragged_buffers[minibatch].resize(config.ragged.size()); in FastParseExample()
1216 size_t start = first_example_of_minibatch(minibatch); in FastParseExample()
1217 size_t end = first_example_of_minibatch(minibatch + 1); in FastParseExample()
1223 status_of_minibatch[minibatch] = FastParseSerializedExample( in FastParseExample()
1227 &varlen_dense_buffers[minibatch], &sparse_buffers[minibatch], in FastParseExample()
[all …]
/external/tensorflow/tensorflow/core/protobuf/tpu/
Doptimization_parameters.proto130 // table, even for entries that are not used in the current minibatch
393 // once per minibatch.
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudnn_6_0.inc1519 cudnnRNNDescriptor_t rnnDesc, const int minibatch,
1526 return func_ptr(rnnDesc, minibatch, dataType, plan);
Dcudnn_7_0.inc1662 cudnnRNNDescriptor_t rnnDesc, const int minibatch,
1669 return func_ptr(rnnDesc, minibatch, dataType, plan);
Dcudnn_7_3.inc1827 cudnnRNNDescriptor_t rnnDesc, const int minibatch,
1834 return func_ptr(rnnDesc, minibatch, dataType, plan);
Dcudnn_7_1.inc1827 cudnnRNNDescriptor_t rnnDesc, const int minibatch,
1834 return func_ptr(rnnDesc, minibatch, dataType, plan);
Dcudnn_7_4.inc1968 cudnnRNNDescriptor_t rnnDesc, const int minibatch,
1975 return func_ptr(rnnDesc, minibatch, dataType, plan);
Dcudnn_8_0.inc1762 cudnnRNNDescriptor_t rnnDesc, const int minibatch,
1769 return func_ptr(rnnDesc, minibatch, dataType, plan);
Dcudnn_7_6.inc2080 cudnnRNNDescriptor_t rnnDesc, const int minibatch,
2087 return func_ptr(rnnDesc, minibatch, dataType, plan);

12