Searched defs:consumer_id (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | outfeed_receiver_test.cc | 59 uint32_t consumer_id; member
63 void Receive(uint32_t consumer_id, std::shared_ptr<Literal> data) { in Receive()
115 std::shared_ptr<Literal> data) { in TEST()
148 std::shared_ptr<Literal> data) { in TEST()
193 std::shared_ptr<Literal> data) { in TEST()
236 std::shared_ptr<Literal> data) { in TEST()
270 std::shared_ptr<Literal> data) { in TEST()
296 std::shared_ptr<Literal> data) { in TEST()
|
| D | outfeed_receiver.cc | 105 OutfeedData(PjRtDevice* device, uint32_t consumer_id, Shape shape) in OutfeedData()
113 uint32_t consumer_id() const { return consumer_id_; } in consumer_id() function in xla::OutfeedData
295 uint32_t consumer_id = header_data[1]; in DeviceListenerThreadLoop() local
389 constexpr int consumer_id = kOutfeedCidShutdown; in SendShutdownOutfeedHeader() local
418 XlaBuilder* builder, XlaOp token, uint32_t consumer_id, in AddOutfeedToBuilder()
468 XlaBuilder* builder, XlaOp token, uint32_t consumer_id, in AddOutfeedToBuilder()
|
| D | outfeed_receiver_py.cc | 52 std::shared_ptr<Literal> literal) { in OutfeedReceiverForPython()
84 uint32_t consumer_id, std::vector<XlaOp> arrays) { in AddOutfeed()
89 void Callback(PjRtDevice* device, uint32_t consumer_id, in Callback()
|
| /external/XNNPACK/src/ |
| D | subgraph.c | 550 const uint32_t consumer_id = value->first_consumer; in xnn_subgraph_optimize() local
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | group_events.cc | 335 absl::optional<uint64> consumer_id; in EventNode() local
|