| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 78 CHECK(LayoutUtil::HasLayout(shape)) in Index()
96 CHECK(LayoutUtil::HasLayout(shape)) in Index()
172 if (linear() != nullptr && LayoutUtil::HasLayout(input_shape) && in SourceIndexOfReshape()
173 LayoutUtil::HasLayout(output_shape) && in SourceIndexOfReshape()
208 if (linear() != nullptr && LayoutUtil::HasLayout(operand_shape) && in SourceIndexOfTranspose()
209 LayoutUtil::HasLayout(shape) && in SourceIndexOfTranspose()
220 CHECK(LayoutUtil::HasLayout(shape) && LayoutUtil::HasLayout(operand_shape)); in SourceIndexOfBitcast()
259 if (linear_ == nullptr || !LayoutUtil::HasLayout(operand_shape) || in SourceIndexOfBroadcast()
260 !LayoutUtil::HasLayout(shape)) { in SourceIndexOfBroadcast()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | optimize_input_output_buffer_alias.cc | 43 TF_RET_CHECK(LayoutUtil::HasLayout(input_shape)); in Build()
44 TF_RET_CHECK(LayoutUtil::HasLayout(output_shape)); in Build()
|
| D | zero_sized_hlo_elimination.cc | 43 if (!LayoutUtil::HasLayout(shape)) { in Run()
|
| D | transfer_manager.cc | 312 if (!LayoutUtil::HasLayout(on_host_shape)) { in AllocateScopedShapedBuffer()
318 TF_RET_CHECK(LayoutUtil::HasLayout(on_device_shape)); in AllocateScopedShapedBuffer()
|
| D | layout_assignment.cc | 406 TF_RET_CHECK(LayoutUtil::HasLayout(data_shape)); in BuildHostChannelConstraints()
774 TF_RET_CHECK(LayoutUtil::HasLayout(shape_with_layout)); in CreateCopyWithNewLayout()
836 TF_RET_CHECK(LayoutUtil::HasLayout(operand->shape())); in CopyOperandIfLayoutsDiffer()
884 TF_RET_CHECK(LayoutUtil::HasLayout(instruction->shape())); in CheckLayouts()
1660 TF_RET_CHECK(LayoutUtil::HasLayout(instruction->shape())); in AssignLayouts()
|
| D | hlo_verifier.cc | 552 TF_RET_CHECK(LayoutUtil::HasLayout(custom_call->shape())); in HandleCustomCall()
562 TF_RET_CHECK(LayoutUtil::HasLayout(operand_shape_with_layout)); in HandleCustomCall()
|
| D | layout_assignment_test.cc | 206 EXPECT_FALSE(LayoutUtil::HasLayout( in TEST_F()
242 EXPECT_TRUE(LayoutUtil::HasLayout(tuple->shape())); in TEST_F()
|
| D | hlo_cost_analysis.cc | 126 if (!LayoutUtil::HasLayout(shape)) { in GetShapeSize()
|
| D | algebraic_simplifier.cc | 2597 if (!LayoutUtil::HasLayout(reshaped_shape)) { in HandleReshape()
3547 TF_RET_CHECK(LayoutUtil::HasLayout(input_shape)); in SimplifyConvToDot()
3548 TF_RET_CHECK(LayoutUtil::HasLayout(filter_shape)); in SimplifyConvToDot()
3549 TF_RET_CHECK(LayoutUtil::HasLayout(convolution_shape)); in SimplifyConvToDot()
|
| D | pattern_matcher.h | 868 return LayoutUtil::HasLayout(*shape) &&
873 if (!LayoutUtil::HasLayout(*shape)) {
|
| D | service.cc | 947 if (!LayoutUtil::HasLayout(return_shape)) { in TransferToClient()
|
| D | hlo_evaluator.cc | 1236 if (!LayoutUtil::HasLayout(instruction->shape())) { in HandleFusion()
|
| D | hlo_parser.cc | 1507 if (!LayoutUtil::HasLayout(shape)) { in ParseInstructionRhs()
1520 if (!LayoutUtil::HasLayout(operand_shape_with_layout)) { in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 306 /* static */ bool LayoutUtil::HasLayout(const Shape& shape) { in HasLayout() function in xla::LayoutUtil
310 [](const Shape& s) { return HasLayout(s); }); in HasLayout()
318 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) { in HasLayout() function in xla::LayoutUtil
320 if (!LayoutUtil::HasLayout(parameter_shape)) { in HasLayout()
324 return LayoutUtil::HasLayout(program_shape.result()); in HasLayout()
|
| D | layout_util.h | 124 static bool HasLayout(const Shape& shape);
127 static bool HasLayout(const ProgramShape& program_shape);
|
| D | literal.cc | 148 CHECK(LayoutUtil::HasLayout(*shape_)); in Literal()
299 if (!LayoutUtil::HasLayout(shape)) { in CreateFromProto()
1172 CHECK(LayoutUtil::HasLayout(literal.shape())); in ToStringHelper()
1173 CHECK(LayoutUtil::HasLayout(subshape)); in ToStringHelper()
1198 CHECK(LayoutUtil::HasLayout(this->shape())); in ToString()
1206 CHECK(LayoutUtil::HasLayout(this->shape())); in ToStringWithoutShape()
1214 CHECK(LayoutUtil::HasLayout(this->shape())); in ToStringWithLayout()
1970 TF_RET_CHECK(LayoutUtil::HasLayout(shape)); in CopyFromProto()
2157 CHECK(LayoutUtil::HasLayout(*shape_)); in MutableBorrowingLiteral()
2168 CHECK(LayoutUtil::HasLayout(*shape_)); in operator =()
[all …]
|
| D | layout_util_test.cc | 269 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
275 EXPECT_FALSE(LayoutUtil::HasLayout(shape)); in TEST_F()
284 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
286 EXPECT_TRUE(LayoutUtil::HasLayout(shape)); in TEST_F()
|
| D | shape_layout.cc | 61 bool ShapeLayout::LayoutIsSet() const { return LayoutUtil::HasLayout(shape_); } in LayoutIsSet()
|
| D | shape_util.cc | 471 } else if (shape.IsArray() && LayoutUtil::HasLayout(shape)) { in HumanStringWithLayout()
686 bool shape_has_valid_layout = LayoutUtil::HasLayout(shape) && in ValidateShapeSize()
1073 CHECK(LayoutUtil::HasLayout(input_shape) && in TransposeIsBitcast()
1074 LayoutUtil::HasLayout(output_shape)); in TransposeIsBitcast()
1104 CHECK(LayoutUtil::HasLayout(input_shape)); in ReshapeIsBitcast()
1105 CHECK(LayoutUtil::HasLayout(output_shape)); in ReshapeIsBitcast()
|
| D | shape.cc | 103 if (LayoutUtil::HasLayout(*this)) { in DeleteDimension()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 30 CHECK(!LayoutUtil::HasLayout(shape) || LayoutUtil::IsDense(shape.layout())); in GetMinimumAlignmentForArray()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 37 } else if (xla::LayoutUtil::HasLayout(shape)) { in PopulateInfeedLayoutVector()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1284 if (!LayoutUtil::HasLayout(shape)) { in Infeed()
1360 if (!LayoutUtil::HasLayout(shape)) { in InfeedWithToken()
1393 if (!LayoutUtil::HasLayout(shape_with_layout)) { in Outfeed()
1446 if (!LayoutUtil::HasLayout(shape_with_layout)) { in OutfeedWithToken()
1509 if (!LayoutUtil::HasLayout(shape)) { in CustomCall()
1523 if (!LayoutUtil::HasLayout(operand_shape)) { in CustomCall()
2410 if (!LayoutUtil::HasLayout(shape_with_layout)) { in SendToHost()
2456 if (!LayoutUtil::HasLayout(shape)) { in RecvFromHost()
|