| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.h | 80 const MutableBorrowingLiteral& literal,
84 const MutableBorrowingLiteral& literal) { in TransferLiteralFromDevice()
102 MutableBorrowingLiteral literal, std::function<void(Status)> done,
106 MutableBorrowingLiteral literal, in TransferLiteralFromDevice()
173 const MutableBorrowingLiteral& literal, std::function<void(Status)> done,
194 MutableBorrowingLiteral literal) = 0;
|
| D | generic_transfer_manager.h | 45 MutableBorrowingLiteral literal, std::function<void(Status)> done,
57 MutableBorrowingLiteral literal) override;
71 MutableBorrowingLiteral literal);
|
| D | generic_transfer_manager.cc | 60 MutableBorrowingLiteral literal, std::function<void(Status)> done, in TransferLiteralFromDevice()
73 MutableBorrowingLiteral literal) { in TransferLiteralFromDeviceInternal()
161 MutableBorrowingLiteral literal) { in TransferLiteralFromOutfeed()
|
| D | transfer_manager.cc | 76 const MutableBorrowingLiteral& literal, in TransferLiteralFromDevice()
179 const MutableBorrowingLiteral& literal, std::function<void(Status)> done, in TransferArrayFromDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | outfeed_manager.h | 42 void set_destination(std::unique_ptr<MutableBorrowingLiteral> destination) { in set_destination()
45 MutableBorrowingLiteral* destination() { return destination_.get(); } in destination()
51 std::unique_ptr<MutableBorrowingLiteral> destination_;
|
| D | gpu_transfer_manager.cc | 149 MutableBorrowingLiteral literal) { in TransferLiteralFromOutfeed()
165 absl::make_unique<MutableBorrowingLiteral>(literal, index)); in TransferLiteralFromOutfeed()
|
| D | gpu_transfer_manager.h | 45 MutableBorrowingLiteral literal) override;
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 744 friend class MutableBorrowingLiteral; variable
797 class MutableBorrowingLiteral : public MutableLiteralBase {
799 virtual ~MutableBorrowingLiteral();
801 MutableBorrowingLiteral() : MutableLiteralBase() {} in MutableBorrowingLiteral() function
803 MutableBorrowingLiteral(const MutableBorrowingLiteral& literal);
804 MutableBorrowingLiteral& operator=(const MutableBorrowingLiteral& literal);
807 MutableBorrowingLiteral(const MutableLiteralBase& literal);
808 MutableBorrowingLiteral(MutableLiteralBase* literal);
809 MutableBorrowingLiteral(MutableBorrowingLiteral literal,
811 MutableBorrowingLiteral(const char* src_buf_ptr, const Shape& shape);
|
| D | literal.cc | 2123 void MutableBorrowingLiteral::CopyPieceSubtree(const Shape& shape, in CopyPieceSubtree()
2153 MutableBorrowingLiteral::MutableBorrowingLiteral( in MutableBorrowingLiteral() function in xla::MutableBorrowingLiteral
2154 const MutableBorrowingLiteral& literal) in MutableBorrowingLiteral()
2165 MutableBorrowingLiteral& MutableBorrowingLiteral::operator=( in operator =()
2166 const MutableBorrowingLiteral& literal) { in operator =()
2178 MutableBorrowingLiteral::MutableBorrowingLiteral( in MutableBorrowingLiteral() function in xla::MutableBorrowingLiteral
2190 MutableBorrowingLiteral::MutableBorrowingLiteral(MutableLiteralBase* literal) in MutableBorrowingLiteral() function in xla::MutableBorrowingLiteral
2201 MutableBorrowingLiteral::MutableBorrowingLiteral( in MutableBorrowingLiteral() function in xla::MutableBorrowingLiteral
2202 MutableBorrowingLiteral literal, const ShapeIndex& view_root) in MutableBorrowingLiteral()
2213 MutableBorrowingLiteral::MutableBorrowingLiteral(const char* src_buf_ptr, in MutableBorrowingLiteral() function in xla::MutableBorrowingLiteral
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util.cc | 42 Tensor* host_tensor, xla::MutableBorrowingLiteral* literal) { in HostTensorToMutableBorrowingLiteral()
51 xla::MutableBorrowingLiteral* literal) { in HostTensorToMutableBorrowingLiteral()
52 *literal = xla::MutableBorrowingLiteral( in HostTensorToMutableBorrowingLiteral()
|
| D | literal_util.h | 41 Tensor* host_tensor, xla::MutableBorrowingLiteral* literal);
47 xla::MutableBorrowingLiteral* literal);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.h | 45 MutableBorrowingLiteral literal) override;
|
| D | cpu_transfer_manager.cc | 179 MutableBorrowingLiteral literal) { in TransferLiteralFromOutfeed()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 238 xla::MutableBorrowingLiteral literal; in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 579 xla::MutableBorrowingLiteral literal; in Compute()
|