Searched refs:ChangeElementType (Results 1 – 15 of 15) 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()
187 ShapeUtil::ChangeElementType(hlo->shape(), replace_with_type_); in Run()
|
| D | dynamic_padder.cc | 154 ShapeUtil::ChangeElementType(operand->shape(), xla::U32); in Run()
156 ShapeUtil::ChangeElementType(operand->shape(), xla::PRED); in Run()
|
| D | reshape_mover.cc | 143 ShapeUtil::ChangeElementType(new_operand_shape, element_type); in UpdateOperand()
234 ShapeUtil::ChangeElementType(new_operand_shape, in PerformSinkReshapeOrTranspose()
|
| D | shape_inference.cc | 283 return ShapeUtil::ChangeElementType( in InferUnaryOpShape()
339 return ShapeUtil::ChangeElementType(shape, PRED); in InferUnaryOpShape()
427 return ShapeUtil::ChangeElementType(operand_shape, new_element_type); in InferConvertShape()
457 return ShapeUtil::ChangeElementType(operand_shape, new_element_type); in InferBitcastConvertShape()
887 return ShapeUtil::ChangeElementType( in InferElementwiseBinaryOpShape()
965 return ShapeUtil::ChangeElementType(shape, C64); in InferBinaryOpShape()
967 return ShapeUtil::ChangeElementType(shape, C128); in InferBinaryOpShape()
988 return ShapeUtil::ChangeElementType(shape, PRED); in InferBinaryOpShape()
1862 Shape result = ShapeUtil::ChangeElementType(in, C64); in InferFftShape()
2757 return ShapeUtil::ChangeElementType(min, operand.element_type()); in InferClampShape()
[all …]
|
| D | bfloat16_normalization.cc | 114 ShapeUtil::ChangeElementType(operand->shape(), to), operand)); in InsertConvertBeforeOperand()
250 ShapeUtil::ChangeElementType(*subshape, BF16), gte)); in HandleMultipleOutputs()
|
| D | algebraic_simplifier.cc | 291 ShapeUtil::ChangeElementType(hlo->shape(), element_type), hlo)); in AsType()
877 ShapeUtil::ChangeElementType(a->shape(), PRED), a, zero_like_a, in TryDivideToShift()
2430 ShapeUtil::ChangeElementType( in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
2443 ShapeUtil::ChangeElementType(operand->shape(), in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
2479 ShapeUtil::ChangeElementType(a->shape(), PRED), a, zero_like_a, in TryRemainderToAnd()
3266 ShapeUtil::ChangeElementType(pad_operand->shape(), in HandleReduceWindow()
|
| D | algebraic_simplifier_test.cc | 979 HloInstruction::CreateBinary(ShapeUtil::ChangeElementType(r2f32, C64), in TEST_F()
1003 HloInstruction::CreateBinary(ShapeUtil::ChangeElementType(r2f32, C64), in TEST_F()
3675 ShapeUtil::ChangeElementType(pad->shape(), F32), pad)); in TEST_F()
|
| D | hlo_instruction.cc | 1122 Shape broadcast_shape = ShapeUtil::ChangeElementType( in CreateBroadcastSequence()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 374 return ChangeElementType(complex_shape, primitive_util::ComplexComponentType( in ComplexComponentShape()
762 /* static */ Shape ShapeUtil::ChangeElementType(const Shape& original, in ChangeElementType() function in xla::ShapeUtil
|
| D | shape_util.h | 338 static Shape ChangeElementType(const Shape& original, PrimitiveType type);
|
| D | literal_comparison.cc | 325 mismatches_ = Literal(ShapeUtil::ChangeElementType(actual_.shape(), PRED)); in Run()
|
| D | literal.cc | 1238 Literal result_literal(ShapeUtil::ChangeElementType( in ConvertBetweenNativeTypesWithConverter()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 122 Shape floating_point_shape = ShapeUtil::ChangeElementType( in PopulateWithComplexData()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 287 xla::ShapeUtil::ChangeElementType(literal.shape(), xla::S64)); in ConstantInputAsInt64Literal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 440 ShapeUtil::ChangeElementType(output_shape, operand_shape.element_type()); in AddBroadcastSequence()
|