Searched defs:crs (Results 1 - 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | bfloat16_normalization_test.cc | 231 HloInstruction* crs = local 235 HloInstruction::CreateGetTupleElement(bf16_shape, crs, 1)); 244 EXPECT_EQ(crs->operand(1)->shape().element_type(), F32); 245 EXPECT_EQ(ShapeUtil::GetSubshape(crs->shape(), {1}).element_type(), F32);
|
H A D | bfloat16_normalization.cc | 38 Status HandleCrossReplicaSum(HloInstruction* crs) override; 148 HloInstruction* crs) { 149 if (!ShapeUtil::IsTuple(crs->shape())) { 150 return HandleInstruction(crs); 153 std::vector<PrimitiveType> operand_types(crs->operand_count()); 154 std::vector<PrimitiveType> output_types(crs->operand_count()); 158 for (int64 i = 0; i < crs->operand_count(); ++i) { 159 operand_types[i] = crs->operand(i)->shape().element_type(); 160 output_types[i] = ShapeUtil::GetSubshape(crs->shape(), {i}).element_type(); 172 for (int64 i = 0; i < crs 147 HandleCrossReplicaSum( HloInstruction* crs) argument [all...] |
H A D | hlo_cost_analysis.cc | 414 Status HloCostAnalysis::HandleCrossReplicaSum(const HloInstruction* crs) { argument 422 crs->shape(), [&, this](const Shape& subshape, const ShapeIndex&) {
|
H A D | hlo_verifier.cc | 92 Status ShapeVerifier::HandleCrossReplicaSum(HloInstruction* crs) { argument 94 for (const HloInstruction* operand : crs->operands()) { 97 return CheckShape(crs,
|
/external/icu/icu4c/source/common/ |
H A D | edits.cpp | 397 Edits::Iterator::Iterator(const uint16_t *a, int32_t len, UBool oc, UBool crs) : argument 399 onlyChanges_(oc), coarse(crs),
|
/external/icu/android_icu4j/src/main/java/android/icu/text/ |
H A D | Edits.java | 234 private Iterator(char[] a, int len, boolean oc, boolean crs) { argument 238 coarse = crs;
|
/external/icu/icu4j/main/classes/core/src/com/ibm/icu/text/ |
H A D | Edits.java | 241 private Iterator(char[] a, int len, boolean oc, boolean crs) { argument 245 coarse = crs;
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | ir_emitter.cc | 618 Status IrEmitter::HandleCrossReplicaSum(HloInstruction* crs) { argument
|
/external/swiftshader/src/Shader/ |
H A D | ShaderCore.cpp | 1210 crs(dst, src1, src2); 1307 void ShaderCore::crs(Vector4f &dst, const Vector4f &src0, const Vector4f &src1) function in class:sw::ShaderCore
|
/external/syslinux/gpxe/src/drivers/net/ |
H A D | via-rhine.c | 462 unsigned long crs:1; member in struct:rhine_tx_desc::VTC_tx_status_tag::__anon24868
|
Completed in 253 milliseconds