Searched defs:crs (Results 1 - 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbfloat16_normalization_test.cc231 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 Dbfloat16_normalization.cc38 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 Dhlo_cost_analysis.cc414 Status HloCostAnalysis::HandleCrossReplicaSum(const HloInstruction* crs) { argument
422 crs->shape(), [&, this](const Shape& subshape, const ShapeIndex&) {
H A Dhlo_verifier.cc92 Status ShapeVerifier::HandleCrossReplicaSum(HloInstruction* crs) { argument
94 for (const HloInstruction* operand : crs->operands()) {
97 return CheckShape(crs,
/external/icu/icu4c/source/common/
H A Dedits.cpp397 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 DEdits.java234 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 DEdits.java241 private Iterator(char[] a, int len, boolean oc, boolean crs) { argument
245 coarse = crs;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc618 Status IrEmitter::HandleCrossReplicaSum(HloInstruction* crs) { argument
/external/swiftshader/src/Shader/
H A DShaderCore.cpp1210 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 Dvia-rhine.c462 unsigned long crs:1; member in struct:rhine_tx_desc::VTC_tx_status_tag::__anon24868

Completed in 253 milliseconds