layout_to_xla_sharding.h
imports tensorflow/core/platform/status.hC declares a character or byte buffer
SHA-256d9e3dcab6579ab2d9e2c3eb92145f32770f1558d7e8df6ebbae8d1d540b41f8a
MaleculeH(Db)Md(In)
Evidence
1/* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is …
11:42… IND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14================================================================ …
17:18… W_DTENSOR_CC_XLA_SPMD_LAYOUT_TO_XLA_SHARDING_H_
18
19#include <cstdint>
20#include <vector>
21
22#include "xla/xla_data.pb.h"
23#include "tensorflow/core/platform/status.h"
24#include "tensorflow/dtensor/cc/dstatus.h"
25#include "tensorflow/dtensor/cc/tensor_layout.h"
26
27namespace tensorflow {
28namespace dtensor {
29
30// Mhlo sharding string attribute, used for sett …
30:71… s, inputs,
31// and outputs of a function for XLA SPMD.
32constexpr char kXlaShardingAttr[] = "mhlo.sharding";
33
34// Represents a permutation of DTensor Layout mesh dimensions from major to
35// minor.
36//
37// Sizes of `perm …