Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_utils.cc36 if (ShapeUtil::IsEffectiveScalar(value->shape())) { in IsValueAllowedInAlternateMemory()
Dhlo_graph_dumper.cc254 .WithShape(m::Shape().IsEffectiveScalar()) in MatchTrivialComputation()
257 .WithShape(m::Shape().IsEffectiveScalar()), in MatchTrivialComputation()
259 .WithShape(m::Shape().IsEffectiveScalar())))) { in MatchTrivialComputation()
991 if (ShapeUtil::IsEffectiveScalar(instr->shape())) { in GetInstructionColor()
1006 ShapeUtil::IsEffectiveScalar(instr->operand(0)->shape())) { in GetInstructionColor()
1021 if (ShapeUtil::IsEffectiveScalar(instr->shape()) || instr->IsFused()) { in GetInstructionColor()
1029 if (ShapeUtil::IsEffectiveScalar(instr->shape())) { in GetInstructionColor()
Dcollective_ops_utils.cc32 .WithShape(m::Shape().IsEffectiveScalar())); in MatchReductionComputation()
Dmulti_output_fusion.cc186 ShapeUtil::IsEffectiveScalar(instr->shape())) { in IsProfitableOperand()
Dwhile_loop_analysis.cc313 if (!ShapeUtil::IsEffectiveScalar(l.shape())) { in LiteralAsScalarInt64()
Dinstruction_fusion.cc217 ShapeUtil::IsEffectiveScalar(operand->shape())) { in EffectivelyAtMostUnary()
Dpattern_matcher.h797 if (!ShapeUtil::IsEffectiveScalar(*shape)) {
1005 constexpr auto IsEffectiveScalar() const {
1730 !ShapeUtil::IsEffectiveScalar(inst->shape())) {
Dhlo_memory_scheduler.cc265 if (ShapeUtil::IsEffectiveScalar(entry.instruction->shape())) { in GetPriority()
Dpattern_matcher_test.cc601 EXPECT_DESC_AND_EXPLANATION(shape, m::Shape().IsEffectiveScalar(), in TEST_F()
Dalgebraic_simplifier.cc156 m::Shape().IsEffectiveScalar())))) { in IsAllFpConstantPowerOf2()
1085 m::Shape().IsEffectiveScalar().WithElementType(PrimitiveType::S32); in TrySimplifyTautologicalCompare()
2516 if (ShapeUtil::IsEffectiveScalar(operand_shape)) { in HandleGather()
3992 if (!ShapeUtil::IsEffectiveScalar(slice->shape())) { in TrySimplifyScalarSlice()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmulti_output_fusion.cc47 ShapeUtil::IsEffectiveScalar(instr->shape())) { in IsProfitableOperand()
Dgpu_fusible.cc286 (!ShapeUtil::IsEffectiveScalar(producer.shape()) || in IsProducerConsumerFusible()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h359 static bool IsEffectiveScalar(const Shape& shape) { in IsEffectiveScalar() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1877 if (ShapeUtil::IsEffectiveScalar(slice->shape())) { in HandleSlice()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc142 ShapeUtil::IsEffectiveScalar(hlo->shape())) { in ReportMemoryUsage()