Searched refs:op_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | zero_sized_hlo_elimination_test.cc | 86 Shape op_shape = ShapeUtil::MakeShape(F32, {4, 0}); in TEST_F() local
87 op_shape.clear_layout(); in TEST_F()
89 HloInstruction::CreateParameter(1, op_shape, "zero sized param 1")); in TEST_F()
91 HloInstruction::CreateParameter(2, op_shape, "zero sized param 2")); in TEST_F()
93 HloInstruction::CreateBinary(op_shape, HloOpcode::kAdd, param1, param2)); in TEST_F()
|
| D | hlo_creation_utils.cc | 388 Shape op_shape = on_true->shape(); in MakeSelectHlo() local
390 if (!ShapeUtil::IsScalar(op_shape) && !op_shape.IsTuple()) { in MakeSelectHlo()
395 ShapeUtil::ChangeElementType(op_shape, PrimitiveType::PRED), pred, in MakeSelectHlo()
403 op_shape.IsTuple() ? HloOpcode::kTupleSelect : HloOpcode::kSelect; in MakeSelectHlo()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression_test.cc | 101 TF_ASSERT_OK_AND_ASSIGN(TensorShape op_shape, in TEST_F()
103 EXPECT_EQ(TensorShape({}), op_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 37 Status CheckSecondToLastDimension(const Shape& op_shape, int64 rank, in CheckSecondToLastDimension() argument
39 const auto actual_num_dims = ShapeUtil::GetDimension(op_shape, rank - 2); in CheckSecondToLastDimension()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 5205 mlir::lmhlo::FusionOp fusion, const Shape& op_shape, in NumInputsInvolveInOnlyElementwiseOps() argument
5210 return ShapeUtil::SameDimensions(op_shape, parameter_shape) && in NumInputsInvolveInOnlyElementwiseOps()
|