Home
last modified time | relevance | path

Searched refs:IsEffectiveScalar (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc279 if (!ShapeUtil::IsEffectiveScalar(root->shape()) || in MatchTrivialComputation()
280 !ShapeUtil::IsEffectiveScalar(operand0->shape()) || in MatchTrivialComputation()
281 !ShapeUtil::IsEffectiveScalar(operand1->shape())) { in MatchTrivialComputation()
923 if (ShapeUtil::IsEffectiveScalar(instr->shape())) { in GetInstructionColor()
936 ShapeUtil::IsEffectiveScalar(instr->operand(0)->shape())) { in GetInstructionColor()
951 if (ShapeUtil::IsEffectiveScalar(instr->shape()) || instr->IsFused()) { in GetInstructionColor()
959 if (ShapeUtil::IsEffectiveScalar(instr->shape())) { in GetInstructionColor()
Dinstruction_fusion.cc149 ShapeUtil::IsEffectiveScalar(operand->shape())) { in EffectivelyUnary()
254 ShapeUtil::IsEffectiveScalar(producer->shape())) { in Run()
Dalgebraic_simplifier.cc720 if (ShapeUtil::IsEffectiveScalar(rhs->shape()) && in HandleDotStrengthReduction()
721 ShapeUtil::IsEffectiveScalar(lhs->shape())) { in HandleDotStrengthReduction()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h279 static bool IsEffectiveScalar(const Shape& shape) { in IsEffectiveScalar() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1870 if (ShapeUtil::IsEffectiveScalar(slice->shape())) { in HandleSlice()