Searched refs:MakeScalarShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 612 ShapeUtil::MakeScalarShape(S32), HloOpcode::kAdd, offsets[concat_dim], in RewriteDynamicConcat()
616 ShapeUtil::MakeScalarShape(S32), HloOpcode::kAdd, offsets[concat_dim], in RewriteDynamicConcat()
662 param_number_before_rewritten, ShapeUtil::MakeScalarShape(PRED), in RewriteDynamicSort()
665 param_number_before_rewritten + 1, ShapeUtil::MakeScalarShape(PRED), in RewriteDynamicSort()
685 ShapeUtil::MakeScalarShape(PRED), HloOpcode::kNot, inbound_rhs)); in RewriteDynamicSort()
688 ShapeUtil::MakeScalarShape(PRED), HloOpcode::kOr, in RewriteDynamicSort()
692 ShapeUtil::MakeScalarShape(PRED), HloOpcode::kAnd, inbound_lhs, in RewriteDynamicSort()
761 ShapeUtil::AppendShapeToTuple(ShapeUtil::MakeScalarShape(S32), in InsertPadToStaticOnInstruction()
|
| D | dynamic_dimension_inference.cc | 186 ShapeUtil::MakeScalarShape(S32), hlo, i + 1)); in HandleCustomCall()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 381 static Shape MakeScalarShape(PrimitiveType element_type);
|
| D | shape_util.cc | 179 /* static */ Shape ShapeUtil::MakeScalarShape(PrimitiveType element_type) { in MakeScalarShape() function in xla::ShapeUtil
|