1 /* Copyright 2018 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 16 #ifndef TENSORFLOW_COMPILER_XLA_SIDE_EFFECT_UTIL_H_ 17 #define TENSORFLOW_COMPILER_XLA_SIDE_EFFECT_UTIL_H_ 18 19 namespace xla { 20 21 // XLA frontend attribute name which specifies TensorFlow rendezvous name. 22 extern const char kXlaHostTransferRendezvousNameAttr[]; 23 24 // XLA frontend attribute name which specifies original host transfer type. 25 // Value is XLA primitive type in lower case. 26 extern const char kXlaHostTransferOriginalTypeAttr[]; 27 28 // XLA frontend attribute name which specifies whether a host transfer 29 // instruction is lower bits for a splitted X64 host transfer. Value is "true" 30 // or "false". 31 extern const char kXlaHostTransferIsLowerBitsAttr[]; 32 33 // XLA frontend attribute name which specifies the name of host side handler 34 // associates with this transfer. 35 extern const char kXlaHostTransferHandlerNameAttr[]; 36 37 // XLA frontend attribute value of the name of TensorFlow Rendezvous Host 38 // Command Handler. 39 extern const char kXlaHostTransferTfRendezvousHandlerName[]; 40 41 // XLA frontend attribute name which specifies the type of computation. 42 extern const char kXlaComputeTypeAttr[]; 43 44 // XLA frontend attribute values for kXlaComputeTypeAttr 45 extern const char kXlaComputeTypeSparse[]; 46 extern const char kXlaComputeTypeDense[]; 47 48 // XLA frontend attribute name for the maximum number of ids expected per 49 // partition *before* an input batch is partitioned. 50 extern const char kXlaMaxIdsPerPartitionAttr[]; 51 52 // XLA frontend attribute name for the maximum number of unique ids expected per 53 // partition *after* an input batch is partitioned. 54 extern const char kXlaMaxUniqueIdsPerPartitionAttr[]; 55 56 // XLA frontend attribute for how to assign ids to partitions. 57 extern const char kXlaShardingStrategyAttr[]; 58 59 // XLA frontend attribute values for kXlaShardingStrategyAttr. 60 extern const char kXlaShardingStrategyMod[]; 61 extern const char kXlaShardingStrategyDiv[]; 62 63 } // namespace xla 64 65 #endif // TENSORFLOW_COMPILER_XLA_SIDE_EFFECT_UTIL_H_ 66