Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc249 .WithShape(m::Shape().IsEffectiveScalar()) in MatchTrivialComputation()
252 .WithShape(m::Shape().IsEffectiveScalar()), in MatchTrivialComputation()
254 .WithShape(m::Shape().IsEffectiveScalar())))) { in MatchTrivialComputation()
971 if (ShapeUtil::IsEffectiveScalar(instr->shape())) { in GetInstructionColor()
986 ShapeUtil::IsEffectiveScalar(instr->operand(0)->shape())) { in GetInstructionColor()
1000 if (ShapeUtil::IsEffectiveScalar(instr->shape()) || instr->IsFused()) { in GetInstructionColor()
1008 if (ShapeUtil::IsEffectiveScalar(instr->shape())) { in GetInstructionColor()
Dmulti_output_fusion.cc138 ShapeUtil::IsEffectiveScalar(instr->shape())) { in IsProfitableOperand()
Dwhile_loop_analysis.cc171 if (!ShapeUtil::IsEffectiveScalar(l.shape())) { in LiteralAsScalarInt64()
Dinstruction_fusion.cc188 ShapeUtil::IsEffectiveScalar(operand->shape())) { in EffectivelyAtMostUnary()
Dpattern_matcher.h816 if (!ShapeUtil::IsEffectiveScalar(*shape)) {
1032 constexpr auto IsEffectiveScalar() const
1744 !ShapeUtil::IsEffectiveScalar(inst->shape())) {
Dalgebraic_simplifier.cc88 m::Shape().IsEffectiveScalar())))) { in IsAllFpConstantPowerOf2()
1215 if (ShapeUtil::IsEffectiveScalar(rhs->shape()) && in HandleDotStrengthReduction()
1216 ShapeUtil::IsEffectiveScalar(lhs->shape())) { in HandleDotStrengthReduction()
2674 if (!ShapeUtil::IsEffectiveScalar(slice->shape())) { in TrySimplifyScalarSlice()
Dpattern_matcher_test.cc600 EXPECT_DESC_AND_EXPLANATION(shape, m::Shape().IsEffectiveScalar(), in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc246 return ShapeUtil::IsEffectiveScalar(producer->shape()) && in ShouldFuseInexpensiveChecks()
Dir_emitter_unnested.cc1384 .WithShape(m::Shape().IsEffectiveScalar())); in IsScalarAddComputation()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h314 static bool IsEffectiveScalar(const Shape& shape) { in IsEffectiveScalar() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1963 if (ShapeUtil::IsEffectiveScalar(slice->shape())) { in HandleSlice()