Searched refs:ValidateLayoutForShape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 195 return ValidateLayoutForShape(shape.layout(), shape); in ValidateLayoutInShape()
207 /* static */ Status LayoutUtil::ValidateLayoutForShape(const Layout& layout, in ValidateLayoutForShape() function in xla::LayoutUtil
387 LayoutUtil::ValidateLayoutForShape(src.layout(), *dst)); in CopyLayoutInternal()
|
| D | packed_literal_reader.cc | 49 LayoutUtil::ValidateLayoutForShape(*layout, literal_shape)); in Read()
|
| D | layout_util.h | 84 static Status ValidateLayoutForShape(const Layout& layout,
|
| D | literal.cc | 727 TF_CHECK_OK(LayoutUtil::ValidateLayoutForShape(new_layout, *subshape)); in Relayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 77 CHECK(LayoutUtil::ValidateLayoutForShape(layout, buffer.shape()).ok()); in BufferLayoutConstraint()
188 LayoutUtil::ValidateLayoutForShape(layout, buffer.shape())); in SetBufferLayout()
1173 LayoutUtil::ValidateLayoutForShape(operand_layout, operand_shape)); in ChooseOperandLayoutFromOutputLayout()
1189 LayoutUtil::ValidateLayoutForShape(operand_layout, operand->shape())); in ChooseOperandLayoutFromOutputLayout()
1234 LayoutUtil::ValidateLayoutForShape(user_layout, output_shape)); in ChooseOutputLayoutFromOperandLayout()
1250 TF_CHECK_OK(LayoutUtil::ValidateLayoutForShape(user_layout, user->shape())); in ChooseOutputLayoutFromOperandLayout()
|
| D | service.cc | 1046 TF_RETURN_IF_ERROR(LayoutUtil::ValidateLayoutForShape( in ComputeConstantGraph()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 150 LayoutUtil::ValidateLayoutForShape(shape.layout(), shape)); in MakeShapeWithLayout()
|