Searched defs:data_handle (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.h | 607 XlaBuilder* builder, XlaOp* data_handle) { in CreateR0Parameter() 620 const std::string& name, XlaBuilder* builder, XlaOp* data_handle) { in CreateR1Parameter() 633 const std::string& name, XlaBuilder* builder, XlaOp* data_handle) { in CreateR2Parameter() 646 const std::string& name, XlaBuilder* builder, XlaOp* data_handle) { in CreateR3Parameter() 659 const std::string& name, XlaBuilder* builder, XlaOp* data_handle) { in CreateR4Parameter() 672 const std::string& name, XlaBuilder* builder, XlaOp* data_handle) { in CreateParameter()
|
| D | compilation_cache_test.cc | 62 auto data_handle = client_ in ExecuteComputationR2F32() local
|
| D | client_library_test_base.cc | 613 XlaBuilder* builder, XlaOp* data_handle) { in CreateParameterAndTransferLiteral() 644 XlaOp* data_handle) { in CreateParameterAndTransferLiteral()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 196 int64 data_handle = 1; field
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1638 RNNDataDescriptor data_handle, in CudnnRnnSequenceTensorDescriptor() 1706 const cudnnRNNDataDescriptor_t data_handle() const { in data_handle() function in stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
|