Searched refs:return_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 818 const Shape* return_shape = &shape; in GetSubshape() local
820 CHECK(IsTuple(*return_shape)) in GetSubshape()
822 return_shape = &return_shape->tuple_shapes(i); in GetSubshape()
824 return *return_shape; in GetSubshape()
829 Shape* return_shape = shape; in GetMutableSubshape() local
831 CHECK(IsTuple(*return_shape)); in GetMutableSubshape()
832 return_shape = return_shape->mutable_tuple_shapes(i); in GetMutableSubshape()
834 return return_shape; in GetMutableSubshape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 246 llvm::Function* function, const Shape& return_shape,
267 llvm::Function* function, const Shape& return_shape, int64 element_count,
|
| D | ir_emitter.cc | 2770 llvm::Function* function, const Shape& return_shape, in EmitElementFunctionCall() argument
2774 function, return_shape, 1, parameter_addresses, name); in EmitElementFunctionCall()
2803 llvm::Function* function, const Shape& return_shape, int64 element_count, in EmitArrayFunctionCall() argument
2808 PrimitiveType return_type = return_shape.element_type(); in EmitArrayFunctionCall()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 1018 const Shape* return_shape; in TransferToClient() local
1023 return_shape = &arg->shape_with_layout(); in TransferToClient()
1025 return_shape = &shaped_buffer->on_host_shape(); in TransferToClient()
1037 if (LayoutUtil::LayoutsInShapesEqual(*return_shape, in TransferToClient()
1042 result_literal->Relayout(*return_shape)->ToProto(); in TransferToClient()
|