Home
last modified time | relevance | path

Searched refs:RandomShuffleQueue (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/
H A Drandom_shuffle_queue_test.py51 q = data_flow_ops.RandomShuffleQueue(10, 5, dtypes_lib.float32)
59 q = data_flow_ops.RandomShuffleQueue(
69 q = data_flow_ops.RandomShuffleQueue(
74 q2 = data_flow_ops.RandomShuffleQueue(
81 q = data_flow_ops.RandomShuffleQueue(
98 q = data_flow_ops.RandomShuffleQueue(10, 0, dtypes_lib.float32)
124 q = data_flow_ops.RandomShuffleQueue(10, 0, dtypes_lib.float32)
148 q = data_flow_ops.RandomShuffleQueue(10, 0, dtypes_lib.float32)
161 q = data_flow_ops.RandomShuffleQueue(3, 0, dtypes_lib.float32)
190 q = data_flow_ops.RandomShuffleQueue(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Drandom_shuffle_queue_op.cc43 class RandomShuffleQueue : public TypedQueue<std::vector<Tensor> > { class
45 RandomShuffleQueue(int32_t capacity, int32_t min_after_dequeue, int64_t seed,
69 ~RandomShuffleQueue() override {} in ~RandomShuffleQueue()
88 TF_DISALLOW_COPY_AND_ASSIGN(RandomShuffleQueue);
91 RandomShuffleQueue::RandomShuffleQueue( in RandomShuffleQueue() function in tensorflow::RandomShuffleQueue
108 Status RandomShuffleQueue::Initialize() { in Initialize()
118 void RandomShuffleQueue::DequeueLocked(OpKernelContext* ctx, Tuple* tuple) { in DequeueLocked()
129 void RandomShuffleQueue::TryEnqueue(const Tuple& tuple, OpKernelContext* ctx, in TryEnqueue()
167 Status RandomShuffleQueue::GetElementComponentFromBatch(const Tuple& tuple, in GetElementComponentFromBatch()
181 void RandomShuffleQueue::TryEnqueueMany(const Tuple& tuple, in TryEnqueueMany()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.queue.-random-shuffle-queue.pbtxt1 path: "tensorflow.queue.RandomShuffleQueue"
3 is_instance: "<class \'tensorflow.python.ops.data_flow_ops.RandomShuffleQueue\'>"
H A Dtensorflow.queue.pbtxt20 name: "RandomShuffleQueue"
H A Dtensorflow.raw_ops.pbtxt3348 name: "RandomShuffleQueue"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-random-shuffle-queue.pbtxt1 path: "tensorflow.RandomShuffleQueue"
3 is_instance: "<class \'tensorflow.python.ops.data_flow_ops.RandomShuffleQueue\'>"
H A Dtensorflow.queue.-random-shuffle-queue.pbtxt1 path: "tensorflow.queue.RandomShuffleQueue"
3 is_instance: "<class \'tensorflow.python.ops.data_flow_ops.RandomShuffleQueue\'>"
H A Dtensorflow.io.-random-shuffle-queue.pbtxt1 path: "tensorflow.io.RandomShuffleQueue"
3 is_instance: "<class \'tensorflow.python.ops.data_flow_ops.RandomShuffleQueue\'>"
H A Dtensorflow.queue.pbtxt20 name: "RandomShuffleQueue"
H A Dtensorflow.io.pbtxt28 name: "RandomShuffleQueue"
H A Dtensorflow.pbtxt200 name: "RandomShuffleQueue"
H A Dtensorflow.raw_ops.pbtxt3348 name: "RandomShuffleQueue"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/python_api/
H A Dapi_def_RandomShuffleQueue.pbtxt2 graph_op_name: "RandomShuffleQueue"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/java_api/
H A Dapi_def_RandomShuffleQueue.pbtxt2 graph_op_name: "RandomShuffleQueue"
H A Dapi_def_RandomShuffleQueueV2.pbtxt4 name: "io.RandomShuffleQueue"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DRandomShuffleQueue.pbtxt2 name: "RandomShuffleQueue"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DRandomShuffleQueue.pbtxt2 name: "RandomShuffleQueue"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_ShuffleAndRepeatDataset.pbtxt9 `RandomShuffleQueue`.
H A Dapi_def_ShuffleDataset.pbtxt9 `RandomShuffleQueue`.
H A Dapi_def_RandomShuffleQueueV2.pbtxt4 name: "RandomShuffleQueue"
H A Dapi_def_RandomShuffleQueue.pbtxt2 graph_op_name: "RandomShuffleQueue"
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dinput.py854 queue = data_flow_ops.RandomShuffleQueue(
896 queue = data_flow_ops.RandomShuffleQueue(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Ddata_flow_ops.py627 class RandomShuffleQueue(QueueBase): class
708 super(RandomShuffleQueue, self).__init__(dtypes, shapes, names, queue_ref)
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties_test.cc655 ops::RandomShuffleQueue(root.WithOpName("Queue2"), {DataType::DT_FLOAT}); in TEST_F()
662 ops::RandomShuffleQueue(root.WithOpName("Queue4"), {DataType::DT_FLOAT}); in TEST_F()
670 auto q5 = ops::RandomShuffleQueue( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/
H A Dops_history.v0.pbtxt22702 name: "RandomShuffleQueue"

12