Searched refs:ValidateLayoutForShape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 193 return ValidateLayoutForShape(shape.layout(), shape); in ValidateLayoutInShape()
205 /* static */ Status LayoutUtil::ValidateLayoutForShape(const Layout& layout, in ValidateLayoutForShape() function in xla::LayoutUtil
397 LayoutUtil::ValidateLayoutForShape(src.layout(), *dst)); in CopyLayoutInternal()
|
| D | packed_literal_reader.cc | 49 LayoutUtil::ValidateLayoutForShape(*layout, literal_shape)); in Read()
|
| D | layout_util.h | 83 static Status ValidateLayoutForShape(const Layout& layout,
|
| D | literal.cc | 583 TF_CHECK_OK(LayoutUtil::ValidateLayoutForShape(new_layout, *subshape)); in Relayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 69 CHECK(LayoutUtil::ValidateLayoutForShape(layout, buffer.shape()).ok()); in BufferLayoutConstraint()
168 LayoutUtil::ValidateLayoutForShape(layout, buffer.shape())); in SetBufferLayout()
1030 LayoutUtil::ValidateLayoutForShape(operand_layout, operand_shape)); in ChooseOperandLayoutFromOutputLayout()
1046 LayoutUtil::ValidateLayoutForShape(operand_layout, operand->shape())); in ChooseOperandLayoutFromOutputLayout()
1091 LayoutUtil::ValidateLayoutForShape(user_layout, output_shape)); in ChooseOutputLayoutFromOperandLayout()
1107 TF_CHECK_OK(LayoutUtil::ValidateLayoutForShape(user_layout, user->shape())); in ChooseOutputLayoutFromOperandLayout()
|
| D | service.cc | 1091 TF_RETURN_IF_ERROR(LayoutUtil::ValidateLayoutForShape( in ComputeConstantGraph()
|