Searched refs:client_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_util.cc | 30 Status ValidateResultShape(const Shape& client_shape, in ValidateResultShape() argument
32 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
33 if (!ShapeUtil::Compatible(client_shape, result_shape)) { in ValidateResultShape()
37 ShapeUtil::HumanStringWithLayout(client_shape), in ValidateResultShape()
|
| D | service.h | 271 Status ValidateResultShape(const Shape& client_shape,
|
| D | service.cc | 218 Status Service::ValidateResultShape(const Shape& client_shape, in ValidateResultShape() argument
220 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
221 if (!ShapeUtil::Compatible(client_shape, result_shape)) { in ValidateResultShape()
225 ShapeUtil::HumanStringWithLayout(client_shape), in ValidateResultShape()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 42 Status ValidateResultShape(const Shape& client_shape, in ValidateResultShape() argument
45 xla::ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
46 if (!xla::ShapeUtil::Compatible(client_shape, result_shape)) { in ValidateResultShape()
50 xla::ShapeUtil::HumanStringWithLayout(client_shape), in ValidateResultShape()
|