Searched refs:data_handle (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.h | 281 XlaBuilder* builder, XlaOp* data_handle);
289 XlaOp* data_handle);
339 XlaOp* data_handle);
352 const string& name, XlaBuilder* builder, XlaOp* data_handle);
366 const string& name, XlaBuilder* builder, XlaOp* data_handle);
380 const string& name, XlaBuilder* builder, XlaOp* data_handle);
394 const string& name, XlaBuilder* builder, XlaOp* data_handle);
401 XlaOp* data_handle);
607 XlaBuilder* builder, XlaOp* data_handle) { in CreateR0Parameter() argument
614 *data_handle = Parameter(builder, parameter_number, literal.shape(), name); in CreateR0Parameter()
[all …]
|
| D | compilation_cache_test.cc | 62 auto data_handle = client_ in ExecuteComputationR2F32() local
66 Literal result = client_->Transfer(*data_handle).ConsumeValueOrDie(); in ExecuteComputationR2F32()
|
| D | client_library_test_base.cc | 613 XlaOp* data_handle) { in CreateParameterAndTransferLiteral() argument
615 nullptr, builder, data_handle); in CreateParameterAndTransferLiteral()
644 XlaOp* data_handle) { in CreateParameterAndTransferLiteral() argument
648 *data_handle = in CreateParameterAndTransferLiteral()
|
| /external/linux-kselftest/tools/testing/selftests/tpm2/ |
| D | tpm2.py | 641 data_handle = struct.unpack('>I', self.send_cmd(cmd)[10:14])[0]
654 data_handle,
661 self.flush_context(data_handle)
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 196 int64 data_handle = 1; field
202 // The inputs of this XRTChainedExecuteOp operation. If data_handle is set,
|
| D | xrt_util.cc | 395 TF_RETURN_IF_ERROR(outputs.Add(i, op.data_handle(), /*release=*/false)); in ExecuteChained()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1459 RNNDataDescriptor data_handle, in CudnnRnnSequenceTensorDescriptor() argument
1466 rnn_data_handle_(std::move(data_handle)), in CudnnRnnSequenceTensorDescriptor()
1529 const cudnnRNNDataDescriptor_t data_handle() const { in data_handle() function in stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1809 /*xDesc=*/input_desc.data_handle(), /*x=*/input_data.opaque(), in DoRnnForwardImpl()
1813 /*yDesc=*/output_desc.data_handle(), in DoRnnForwardImpl()
1840 /*xDesc=*/input_desc.data_handle(), /*x=*/input_data.opaque(), in DoRnnForwardImpl()
1844 /*yDesc=*/output_desc.data_handle(), in DoRnnForwardImpl()
1939 /*yDesc=*/output_desc.data_handle(), /*y=*/output_data.opaque(), in DoRnnBackwardImpl()
1940 /*dyDesc=*/output_desc.data_handle(), in DoRnnBackwardImpl()
1949 /*dxDesc=*/input_desc.data_handle(), in DoRnnBackwardImpl()
[all …]
|