Searched refs:ChangeElementType (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_element_type_converter.cc | 40 Shape shape = ShapeUtil::ChangeElementType(hlo->shape(), type); in ToElementType()
73 subshape = ShapeUtil::ChangeElementType(subshape, to_type); in GetConvertedTupleShape()
180 ShapeUtil::ChangeElementType(hlo->shape(), replace_with_type_); in Run()
|
| D | reshape_mover.cc | 197 ShapeUtil::ChangeElementType(new_operand_shape, element_type); in UpdateOperand()
297 ShapeUtil::ChangeElementType(new_operand_shape, in TrySinkReshapeOrTranspose()
|
| D | shape_inference.cc | 349 return ShapeUtil::ChangeElementType(arg, F32); in InferUnaryOpShape()
352 return ShapeUtil::ChangeElementType( in InferUnaryOpShape()
379 return ShapeUtil::ChangeElementType(arg, PRED); in InferUnaryOpShape()
469 return ShapeUtil::ChangeElementType(operand_shape, new_element_type); in InferConvertShape()
499 return ShapeUtil::ChangeElementType(operand_shape, new_element_type); in InferBitcastConvertShape()
915 return ShapeUtil::ChangeElementType( in InferElementwiseBinaryOpShape()
990 return ShapeUtil::ChangeElementType(shape, C64); in InferBinaryOpShape()
1015 return ShapeUtil::ChangeElementType(shape, PRED); in InferBinaryOpShape()
1788 Shape result = ShapeUtil::ChangeElementType(in, C64); in InferFftShape()
2367 return ShapeUtil::ChangeElementType(min, operand.element_type()); in InferClampShape()
[all …]
|
| D | bfloat16_normalization.cc | 108 ShapeUtil::ChangeElementType(operand->shape(), to), operand)); in InsertConvertBeforeOperand()
207 ShapeUtil::ChangeElementType(*subshape, BF16), gte)); in HandleCrossReplicaSum()
|
| D | algebraic_simplifier_test.cc | 533 HloInstruction::CreateBinary(ShapeUtil::ChangeElementType(r2f32, C64), in TEST_F()
557 HloInstruction::CreateBinary(ShapeUtil::ChangeElementType(r2f32, C64), in TEST_F()
|
| D | hlo_instruction.cc | 682 Shape broadcast_shape = ShapeUtil::ChangeElementType( in CreateBroadcastSequence()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 300 static Shape ChangeElementType(const Shape& original, PrimitiveType type);
|
| D | shape_util.cc | 362 return ChangeElementType(complex_shape, primitive_util::ComplexComponentType( in ComplexComponentShape()
809 /* static */ Shape ShapeUtil::ChangeElementType(const Shape& original, in ChangeElementType() function in xla::ShapeUtil
|
| D | literal_util.cc | 1291 auto result_literal = MakeUnique<Literal>(ShapeUtil::ChangeElementType( in ConvertBetweenNativeTypes()
1308 ShapeUtil::ChangeElementType(src_literal.shape(), C64)); in ConvertToC64()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 234 xla::ShapeUtil::ChangeElementType(literal.shape(), xla::S64)); in ConstantInputAsInt64Literal()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | literal_test_util.cc | 383 miscompares_ = Literal(ShapeUtil::ChangeElementType(actual.shape(), PRED)); in ExpectNear()
|