| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.h | 45 static HloSharding Replicate(absl::Span<const OpMetadata> metadata = {}) {
50 static HloSharding Manual(absl::Span<const OpMetadata> metadata = {}) {
57 absl::Span<const OpMetadata> metadata = {});
62 absl::Span<const OpMetadata> metadata = {}) {
73 absl::Span<const OpMetadata> metadata = {});
80 absl::Span<const OpMetadata> metadata = {});
85 absl::Span<const OpMetadata> metadata = {});
240 HloSharding WithMetadata(absl::Span<const OpMetadata> metadata,
288 std::vector<OpMetadata>& metadata() { return metadata_; } in metadata()
289 const std::vector<OpMetadata>& metadata() const { return metadata_; } in metadata()
[all …]
|
| D | hlo_sharding_test.cc | 42 OpMetadata GetMetadata(const std::string& op_name) { in GetMetadata()
43 OpMetadata metadata; in GetMetadata()
48 std::vector<OpMetadata> SingleMetadata() { return {GetMetadata("a")}; } in SingleMetadata()
50 std::vector<OpMetadata> ListMetadata() { in ListMetadata()
271 std::tuple<std::vector<OpMetadata>, std::string>;
291 std::make_tuple(std::vector<OpMetadata>(), "{replicated}"),
317 std::make_tuple(std::vector<OpMetadata>(), "{maximal device=7}"),
345 std::make_tuple(std::vector<OpMetadata>(), "{devices=[2,1,2]2,3,5,7}"),
388 : public ::testing::TestWithParam<std::vector<OpMetadata>> {};
435 ::testing::Values(std::vector<OpMetadata>(),
|
| D | source_map_util.h | 29 Status InvalidParameterArgument(const OpMetadata& op_metadata, in InvalidParameterArgument()
55 const OpMetadata& metadata = param->metadata(); in InvalidParameterArgument()
|
| D | local_service.cc | 80 absl::optional<const OpMetadata*> ParameterMetadata( in ParameterMetadata()
121 absl::optional<const OpMetadata*> metadata = in CompileExecutables()
128 const OpMetadata& m = *metadata.value(); in CompileExecutables()
|
| D | hlo_op_metadata.h | 24 std::string OpMetadataToString(const OpMetadata& metadata);
|
| D | hlo_sharding.cc | 32 absl::Span<const OpMetadata> metadata) { in AssignDevice()
37 absl::Span<const OpMetadata> metadata) { in Tile1D()
50 absl::Span<const OpMetadata> metadata) { in PartialTile()
69 absl::Span<const OpMetadata> metadata) { in PartialTile()
475 std::vector<OpMetadata> metadata(proto.metadata().begin(), in FromProto()
662 HloSharding HloSharding::WithMetadata(absl::Span<const OpMetadata> metadata, in WithMetadata()
|
| D | hlo_op_metadata.cc | 26 std::string OpMetadataToString(const OpMetadata& metadata) { in OpMetadataToString()
|
| D | while_util.h | 87 const OpMetadata& metadata);
|
| D | gather_expander_test.cc | 134 OpMetadata metadata; in TEST_F()
|
| D | hlo_computation.h | 109 MetadataBuilder(HloComputation* computation, const OpMetadata& metadata) in MetadataBuilder()
120 OpMetadata metadata_;
|
| D | sharding_propagation_test.cc | 39 instruction->set_metadata(OpMetadata()); in ClearMetadata()
62 std::string OpMetadataListToString(absl::Span<const OpMetadata> metadata) { in OpMetadataListToString()
65 for (const OpMetadata& element : metadata) { in OpMetadataListToString()
75 explicit HloShardingMetadataMatcher(absl::Span<const OpMetadata> metadata) in HloShardingMetadataMatcher()
106 std::vector<OpMetadata> metadata_;
110 absl::Span<const OpMetadata> metadata) { in ShardingMetadata()
114 OpMetadata CreateMetadata(const std::string& op_name) { in CreateMetadata()
115 OpMetadata metadata; in CreateMetadata()
1660 absl::Span<const absl::Span<const OpMetadata>> in TEST_P()
2567 const OpMetadata& metadata) { in TEST_P()
|
| D | while_util.cc | 243 const OpMetadata& metadata) { in MakeCountedLoop()
|
| D | hlo_instruction.h | 1606 void set_metadata(const OpMetadata& metadata) { in set_metadata()
1620 const OpMetadata& metadata() const { return metadata_; } in metadata()
2137 OpMetadata metadata_;
|
| D | hlo_sharding_util.cc | 176 std::vector<OpMetadata> merged_metadata(std::move(to_merge->metadata())); in MergeSharding()
178 const absl::flat_hash_set<OpMetadata, protobuf_util::ProtobufHashWrapper, in MergeSharding()
182 [&metadata_set](const OpMetadata& data) { in MergeSharding()
|
| D | hlo_parser.cc | 410 bool ParseMetadata(OpMetadata* metadata);
412 tensorflow::protobuf::RepeatedPtrField<OpMetadata>* metadata);
1011 optional<OpMetadata> metadata; in ParseInstructionRhs()
3773 OpMetadata result; in ParseAttributeHelper()
3777 static_cast<optional<OpMetadata>*>(attr_out_ptr)->emplace(result); in ParseAttributeHelper()
4748 bool HloParserImpl::ParseMetadata(OpMetadata* metadata) { in ParseMetadata()
4789 tensorflow::protobuf::RepeatedPtrField<OpMetadata>* metadata) { in ParseSingleOrListMetadata()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util.cc | 29 xla::OpMetadata CreateOpMetadata(const std::string& op_type, in CreateOpMetadata()
31 xla::OpMetadata metadata; in CreateOpMetadata()
59 absl::optional<xla::OpMetadata> metadata) { in ParseShardingFromDevice()
95 add_metadata ? absl::optional<xla::OpMetadata>( in ParseShardingFromDevice()
110 add_metadata ? absl::optional<xla::OpMetadata>( in ParseShardingFromDevice()
|
| D | xla_compilation_device.cc | 90 static void AttachLocationToMetadata(xla::OpMetadata& metadata, in AttachLocationToMetadata()
107 xla::OpMetadata metadata; in Compute()
|
| D | sharding_util.h | 39 absl::optional<xla::OpMetadata> metadata = absl::nullopt);
|
| D | xla_compiler.cc | 182 xla::OpMetadata retval_metadata; in BuildComputation()
1047 xla::OpMetadata arg_metadata; in BuildArguments()
1088 xla::OpMetadata arg_metadata; in BuildArguments()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_metadata_test.cc | 40 OpMetadata metadata_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 52 const OpMetadata& metadata) { in CreateCholesky()
|
| D | gpu_conv_rewriter_test.cc | 120 OpMetadata metadata; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 162 class OpMetadata(object): class
177 return OpMetadata(
|
| D | types.h | 388 struct type_caster<xla::OpMetadata> {
390 PYBIND11_TYPE_CASTER(xla::OpMetadata, _("xla::OpMetadata"));
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 174 void SetOpMetadata(OpMetadata metadata) { metadata_ = std::move(metadata); } in SetOpMetadata()
179 OpMetadata SwapOpMetadata(OpMetadata metadata) { in SwapOpMetadata()
180 OpMetadata old_metadata = std::move(metadata_); in SwapOpMetadata()
186 void SetOneShotOpMetadata(OpMetadata metadata) { in SetOneShotOpMetadata()
1036 OpMetadata metadata_;
1039 absl::optional<OpMetadata> one_shot_metadata_;
1533 XlaScopedOpMetadataAssignment(xla::XlaBuilder* builder, OpMetadata metadata) in XlaScopedOpMetadataAssignment()
1542 OpMetadata saved_;
|