Home
last modified time | relevance | path

Searched refs:add_metadata (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc42 *sharding_element.add_metadata() = metadata; in AssignOpMetadataToSharding()
45 *sharding.add_metadata() = metadata; in AssignOpMetadataToSharding()
82 *sharding.add_metadata() = metadata.value(); in ParseShardingFromDevice()
89 const NodeDef& node_def, int num_cores_per_replica, bool add_metadata) { in ParseShardingFromDevice() argument
92 GetShardingFromNodeDef(node_def, add_metadata)); in ParseShardingFromDevice()
95 add_metadata ? absl::optional<xla::OpMetadata>( in ParseShardingFromDevice()
101 const Node& node, int num_cores_per_replica, bool add_metadata) { in ParseShardingFromDevice() argument
107 GetShardingFromNodeDef(node.def(), add_metadata)); in ParseShardingFromDevice()
110 add_metadata ? absl::optional<xla::OpMetadata>( in ParseShardingFromDevice()
116 const Edge& edge, int num_cores_per_replica, bool add_metadata) { in ParseShardingFromEdgeSource() argument
[all …]
Dsharding_util.h42 const Node& node, int num_cores_per_replica, bool add_metadata);
45 const NodeDef& node_def, int num_cores_per_replica, bool add_metadata);
48 const Edge& edge, int num_cores_per_replica, bool add_metadata);
54 const NodeDef& node_def, bool add_metadata);
Dxla_compiler.cc1447 tf2xla::TensorMetadata* metadata = transfer->add_metadata(); in SetTransfer()
/external/rust/crates/grpcio-sys/grpc/src/core/lib/surface/
Dcall_log_batch.cc35 static void add_metadata(const grpc_metadata* md, size_t count, in add_metadata() function
56 add_metadata(op->data.send_initial_metadata.metadata, in grpc_op_string()
78 add_metadata(op->data.send_status_from_server.trailing_metadata, in grpc_op_string()
/external/grpc-grpc/src/core/lib/surface/
Dcall_log_batch.cc30 static void add_metadata(gpr_strvec* b, const grpc_metadata* md, size_t count) { in add_metadata() function
56 add_metadata(&b, op->data.send_initial_metadata.metadata, in grpc_op_string()
78 add_metadata(&b, op->data.send_status_from_server.trailing_metadata, in grpc_op_string()
/external/rust/crates/grpcio/src/
Dmetadata.rs86 Ok(self.add_metadata(&key, value.as_bytes())) in add_str()
89 fn add_metadata(&mut self, key: &str, value: &[u8]) -> &mut MetadataBuilder { in add_metadata() method
107 Ok(self.add_metadata(&key, value)) in add_bytes()
224 builder.add_metadata(k, v); in clone()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_test.cc102 *tiled->add_metadata() = GetMetadata("a"); in TEST_F()
103 *tiled->add_metadata() = GetMetadata("b"); in TEST_F()
106 *replicated->add_metadata() = GetMetadata("c"); in TEST_F()
Dhlo_sharding.cc543 *result.add_metadata() = metadata; in ToProto()
/external/perfetto/src/trace_processor/importers/proto/
Dproto_trace_parser_unittest.cc2454 auto* metadata = bundle->add_metadata(); in TEST_F()
2483 auto* metadata = bundle->add_metadata(); in TEST_F()
2486 metadata = bundle->add_metadata(); in TEST_F()
2525 auto* metadata = bundle->add_metadata(); in TEST_F()
2661 auto* metadata = track_event->add_metadata(); in TEST_F()
2667 auto* metadata = track_event->add_metadata(); in TEST_F()
2673 auto* metadata = track_event->add_metadata(); in TEST_F()
2679 auto* metadata = track_event->add_metadata(); in TEST_F()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1234 *core_zero_sharding.add_metadata() = in ParseAndValidateSharding()
2066 *node_and_sharding->sharding.add_metadata() = in AssignArgsAndRetvalsToCores()
2179 *node_and_sharding->sharding.add_metadata() = in AssignArgsAndRetvalsToCores()
/external/tensorflow/tensorflow/lite/schema/
Dschema_generated.h11401 …void add_metadata(flatbuffers::Offset<flatbuffers::Vector<flatbuffers::Offset<tflite::Metadata>>> …
11431 builder_.add_metadata(metadata);