Searched refs:FromC (Results 1 – 10 of 10) sorted by relevance
42 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base);46 xla::Shape FromC(const XLA_Shape* c_shape);51 xla::Layout FromC(const XLA_Layout* c_layout);56 xla::Tile FromC(const XLA_Tile* c_tile);62 xla::ShapeIndex FromC(XLA_ShapeIndex* c_shape);67 xla::MutableBorrowingLiteral FromC(XLA_Literal* c_literal);72 xla::ShapedBuffer FromC(XLA_ShapedBuffer* c_buffer);77 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base);82 xla::MutableBorrowingLiteral FromC(XLA_Literal* c_literal);87 xla::ShapedBuffer FromC(XLA_ShapedBuffer* c_buffer);[all …]
25 xla::ShapedBuffer FromC(XLA_ShapedBuffer* c_buffer) { in FromC() function27 ApiConverter::FromC(&c_buffer->on_device_shape); in FromC()33 pair.second = ApiConverter::FromC(c_buffer->bases[i]); in FromC()65 xla::MaybeOwningDeviceMemory FromC( in FromC() function70 stream_executor::OwningDeviceMemory(ApiConverter::FromC(se_mem->memory), in FromC()73 return xla::MaybeOwningDeviceMemory(ApiConverter::FromC(se_mem->memory)); in FromC()112 ->Deallocate(device_ordinal, ApiConverter::FromC(*base)); in ToC()144 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base) { in FromC() function233 xla::Shape FromC(const XLA_Shape* c_shape) { in FromC() function241 tuple_shapes.push_back(FromC(&c_shape->tuple_shapes[i])); in FromC()[all …]
32 static tensorflow::Status FromC(TF_Status* const c_status) { in FromC() function47 tensorflow::Status status() const { return FromC(c_status); } in status()
61 xla::Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape()179 done(StatusHelper::FromC(overall_status)); in TransferFinished()238 xla::Shape output = ApiConverter::FromC(&c_output); in ChooseCompactLayoutForShape()
112 Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape()
198 return ApiConverter::FromC(se_base); in Allocate()
156 ApiConverter::FromC(&se_execution_output.result), in ExecuteAsyncOnStream()163 ApiConverter::FromC(&se_execution_output.aliased_indices[i])); in ExecuteAsyncOnStream()170 ApiConverter::FromC(&se_execution_output.to_be_released[i], in ExecuteAsyncOnStream()336 ApiConverter::FromC(c_module)); in Compile()
112 xla::Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape()
42 xla::Shape infeed_shape = ApiConverter::FromC(&c_infeed_shape); in GetTPUInfeedLayout()
65 xla::Shape infeed_shape = ApiConverter::FromC(&c_infeed_shape); in GetTPUInfeedLayout()