/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_AddManySparseToTensorsMap.pbtxt | 6 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
|
D | api_def_SerializeManySparse.pbtxt | 6 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]`.
|
D | api_def_TakeManySparseFromTensorsMap.pbtxt | 13 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
|
D | api_def_DeserializeManySparse.pbtxt | 16 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
|
D | api_def_ParseExample.pbtxt | 71 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
|
D | api_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt | 13 This is the label for the given minibatch entry.
|
D | api_def_ParseExampleV2.pbtxt | 102 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
|
D | api_def_DecodeProtoV2.pbtxt | 77 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`
|
D | api_def_AddSparseToTensorsMap.pbtxt | 50 The `SparseTensor` can then be read out as part of a minibatch by passing
|
D | api_def_PaddingFIFOQueue.pbtxt | 54 size of any given element in the minibatch. See below for details.
|
D | api_def_PaddingFIFOQueueV2.pbtxt | 56 size of any given element in the minibatch. See below for details.
|
D | api_def_ExperimentalParseExampleDataset.pbtxt | 52 minibatch elements smaller than the maximum number of blocks for the
|
D | api_def_ParseExampleDataset.pbtxt | 52 minibatch elements smaller than the maximum number of blocks for the
|
D | api_def_ParseExampleDatasetV2.pbtxt | 52 minibatch elements smaller than the maximum number of blocks for the
|
/external/tensorflow/tensorflow/core/kernels/ |
D | serialize_sparse_op.cc | 149 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()
|
D | sparse_tensors_map_ops.cc | 282 sparse::GroupIterable minibatch = input_st.group({0}); in Compute() local 283 for (const auto& subset : minibatch) { in Compute()
|
/external/tensorflow/tensorflow/core/util/ |
D | example_proto_fast_parsing.cc | 1195 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/ |
D | optimization_parameters.proto | 130 // table, even for entries that are not used in the current minibatch 393 // once per minibatch.
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cudnn_6_0.inc | 1519 cudnnRNNDescriptor_t rnnDesc, const int minibatch, 1526 return func_ptr(rnnDesc, minibatch, dataType, plan);
|
D | cudnn_7_0.inc | 1662 cudnnRNNDescriptor_t rnnDesc, const int minibatch, 1669 return func_ptr(rnnDesc, minibatch, dataType, plan);
|
D | cudnn_7_3.inc | 1827 cudnnRNNDescriptor_t rnnDesc, const int minibatch, 1834 return func_ptr(rnnDesc, minibatch, dataType, plan);
|
D | cudnn_7_1.inc | 1827 cudnnRNNDescriptor_t rnnDesc, const int minibatch, 1834 return func_ptr(rnnDesc, minibatch, dataType, plan);
|
D | cudnn_7_4.inc | 1968 cudnnRNNDescriptor_t rnnDesc, const int minibatch, 1975 return func_ptr(rnnDesc, minibatch, dataType, plan);
|
D | cudnn_8_0.inc | 1762 cudnnRNNDescriptor_t rnnDesc, const int minibatch, 1769 return func_ptr(rnnDesc, minibatch, dataType, plan);
|
D | cudnn_7_6.inc | 2080 cudnnRNNDescriptor_t rnnDesc, const int minibatch, 2087 return func_ptr(rnnDesc, minibatch, dataType, plan);
|