Home
last modified time | relevance | path

Searched refs:ExecuteParams (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmemset_thunk.cc24 Status MemzeroThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
33 Status Memset32BitValueThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
Dcudnn_batchnorm_thunk.h64 Status ExecuteOnStream(const ExecuteParams& params) override;
92 Status ExecuteOnStream(const ExecuteParams& params) override;
120 Status ExecuteOnStream(const ExecuteParams& params) override;
Dmemset_thunk.h39 Status ExecuteOnStream(const ExecuteParams& params) override;
55 Status ExecuteOnStream(const ExecuteParams& params) override;
Dcopy_thunk.h43 Status ExecuteOnStream(const ExecuteParams& params) override;
65 Status ExecuteOnStream(const ExecuteParams& params) override;
Dcopy_thunk.cc32 Status HostToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
49 Status DeviceToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
Dnccl_collective_thunk.h123 Status ExecuteOnStream(const ExecuteParams& params) override;
126 virtual Status RunNcclCollective(const ExecuteParams& params,
Dcudnn_batchnorm_thunk.cc51 const ExecuteParams& params) { in ExecuteOnStream()
90 const ExecuteParams& params) { in ExecuteOnStream()
138 const ExecuteParams& params) { in ExecuteOnStream()
Dthunk.h105 struct ExecuteParams { struct
123 virtual Status ExecuteOnStream(const ExecuteParams& params) = 0; argument
Dinfeed_thunk.h42 Status ExecuteOnStream(const ExecuteParams& params) override;
Doutfeed_thunk.h40 Status ExecuteOnStream(const ExecuteParams& params) override;
Dreplica_id_thunk.cc21 Status ReplicaOrPartitionIdThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
Dcollective_permute_thunk.h35 Status ExecuteOnStream(const ExecuteParams& params) override;
Dcustom_call_thunk.h46 Status ExecuteOnStream(const ExecuteParams& params) override;
Dconvolution_thunk.h53 Status ExecuteOnStream(const ExecuteParams& params) override;
Dfor_thunk.h42 Status ExecuteOnStream(const ExecuteParams& params) override;
Dnccl_all_reduce_thunk.h48 Status RunNcclCollective(const ExecuteParams& params,
Dsequential_thunk.h44 Status ExecuteOnStream(const ExecuteParams& params) override;
Dnccl_all_to_all_thunk.h48 Status RunNcclCollective(const ExecuteParams& params,
Dnccl_all_gather_thunk.h50 Status RunNcclCollective(const ExecuteParams& params,
Dsequential_thunk.cc39 Status SequentialThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
Dreplica_id_thunk.h28 Status ExecuteOnStream(const ExecuteParams& params) override;
Dwhile_thunk.h53 Status ExecuteOnStream(const ExecuteParams& params) override;
Dkernel_thunk.h62 Status ExecuteOnStream(const ExecuteParams& params) override;
Dfor_thunk.cc44 Status ForThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
Dcholesky_thunk.h53 Status ExecuteOnStream(const ExecuteParams& params) override;

12