Searched defs:consumer_id (Results 1 – 10 of 10) 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()
116 std::shared_ptr<Literal> data) { in TEST()
149 std::shared_ptr<Literal> data) { in TEST()
194 std::shared_ptr<Literal> data) { in TEST()
237 std::shared_ptr<Literal> data) { in TEST()
271 std::shared_ptr<Literal> data) { in TEST()
297 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
297 uint32_t consumer_id = header_data[1]; in DeviceListenerThreadLoop() local
396 constexpr int consumer_id = kOutfeedCidShutdown; in SendShutdownOutfeedHeader() local
425 XlaBuilder* builder, XlaOp token, uint32_t consumer_id, in AddOutfeedToBuilder()
479 XlaBuilder* builder, XlaOp token, uint32_t consumer_id, in AddOutfeedToBuilder()
|
| D | outfeed_receiver_py.cc | 51 std::shared_ptr<Literal> literal) { in OutfeedReceiverForPython()
83 uint32_t consumer_id, std::vector<XlaOp> arrays) { in AddOutfeed()
88 void Callback(PjRtDevice* device, uint32_t consumer_id, in Callback()
|
| /external/perfetto/src/trace_processor/importers/proto/ |
| D | android_probes_tracker.h | 84 int32_t consumer_id) { in GetEnergyBreakdownDescriptor()
93 void SetEnergyBreakdownDescriptor(int32_t consumer_id, in SetEnergyBreakdownDescriptor()
|
| D | android_probes_parser.cc | 167 auto consumer_id = event.energy_consumer_id(); in ParseEnergyBreakdown() local
|
| /external/ComputeLibrary/arm_compute/graph/ |
| D | Edge.h | 76 NodeID consumer_id() const in consumer_id() function
|
| /external/perfetto/src/trace_processor/importers/common/ |
| D | track_tracker.cc | 343 int32_t consumer_id, in InternEnergyCounterTrack()
375 int32_t consumer_id, in InternEnergyPerUidCounterTrack()
|
| /external/XNNPACK/src/ |
| D | subgraph.c | 973 const uint32_t consumer_id = value->first_consumer; in xnn_subgraph_fusion() local
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | group_events.cc | 109 std::optional<uint64_t> consumer_id; member
|
| D | xplane_utils.cc | 380 absl::optional<uint64_t> consumer_id; in AddFlowsToXplane() local
|