| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | sequential_thunk.cc | 37 for (auto& thunk : thunks_) { in Initialize() local
44 for (const auto& thunk : thunks_) { in ExecuteOnStream() local
|
| D | ir_emitter_unnested.h | 234 void AddThunkToThunkSequence(std::unique_ptr<Thunk> thunk) { in AddThunkToThunkSequence()
|
| D | ir_emitter_unnested.cc | 1108 auto thunk = in EmitGemmThunk() local
1133 auto thunk = std::make_unique<CublasLtMatmulThunk>( in EmitCublasLtMatmulThunk() local
1366 auto thunk = std::make_unique<CustomCallThunk>( in EmitCustomCallThunk() local
2279 Thunk* thunk, mlir::lmhlo::ScatterOp scatter, in EmitScatter()
2308 const ScatterDescriptor& desc, Thunk* thunk, in EmitScatter()
2754 auto thunk = std::make_unique<ThunkType>(GetThunkInfo(op), result_slice); in EmitReplicaOrPartitionId() local
2784 auto thunk = std::make_unique<NcclCollectivePermuteThunk>( in EmitCollectivePermute() local
2795 mlir::Operation* op, Thunk* thunk) { in MaybeAddAllReduceStartThunkToMap()
2840 auto thunk = in EmitNcclThunk() local
2938 auto thunk = in EmitInfeed() local
[all …]
|
| D | gpu_executable.cc | 80 bool NeedsAsyncCommsStream(Thunk& thunk) { in NeedsAsyncCommsStream()
335 for (const std::unique_ptr<Thunk>& thunk : thunk_sequence) { in ExecuteThunks() local
|
| /external/escapevelocity/src/main/java/com/google/escapevelocity/ |
| D | Macro.java | 101 Node thunk = parameterThunks.get(var); in getVar() local
123 final Node thunk = parameterThunks.get(var); in setVar() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_cord_test.cc | 81 auto thunk = CreateThunkFor(cleaner); in TEST() local
148 auto thunk = CreateThunkFor(cleaner); in TEST() local
169 auto thunk = CreateThunkFor(cleaner); in TEST() local
189 auto thunk = CreateThunkFor(cleaner); in TEST() local
|
| /external/tensorflow/tensorflow/python/autograph/core/ |
| D | function_wrappers.py | 110 def with_function_scope(thunk, scope_name, options): argument
|
| /external/grpc-grpc/src/python/grpcio/grpc/ |
| D | _interceptor.py | 696 thunk = lambda m: self._channel.unary_unary( function
717 thunk = lambda m: self._channel.unary_stream( function
738 thunk = lambda m: self._channel.stream_unary( function
759 thunk = lambda m: self._channel.stream_stream( function
|
| /external/cronet/stable/base/win/ |
| D | iat_patch_function.cc | 39 IMAGE_THUNK_DATA thunk; in GetIATFunction() member
|
| /external/cronet/tot/base/win/ |
| D | iat_patch_function.cc | 39 IMAGE_THUNK_DATA thunk; in GetIATFunction() member
|
| /external/rust/android-crates-io/crates/quiche/deps/boringssl/src/crypto/fipsmodule/rand/ |
| D | urandom_test.cc | 154 std::function<void()> thunk) { in GetTrace()
|
| /external/pytorch/test/functorch/ |
| D | common_utils.py | 584 def check_vmap_fallback(test_case, thunk, opinfo, dry_run=False): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| D | kernel_ops_pattern.cc | 300 if (!llvm::all_of(*thunks, [](const auto& thunk) { in Match()
|
| /external/boringssl/src/crypto/rand_extra/ |
| D | urandom_test.cc | 389 std::function<void()> thunk) { in GetTrace()
|
| /external/cronet/stable/third_party/boringssl/src/crypto/rand_extra/ |
| D | urandom_test.cc | 408 std::function<void()> thunk) { in GetTrace()
|
| /external/cronet/tot/third_party/boringssl/src/crypto/rand_extra/ |
| D | urandom_test.cc | 408 std::function<void()> thunk) { in GetTrace()
|
| /external/python/cpython3/Modules/_ctypes/ |
| D | ctypes.h | 181 CThunkObject *thunk; member
|
| D | _ctypes.c | 3886 CThunkObject *thunk; in PyCFuncPtr_new() local
|
| /external/perfetto/src/trace_processor/importers/proto/ |
| D | heap_graph_tracker.cc | 682 ObjectTable::Id thunk; in PopulateNativeSize() member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 1034 def __init__(self, thunk): argument
|
| /external/llvm/include/llvm/Support/ |
| D | MachO.h | 858 uint32_t thunk; member
864 uint64_t thunk; member
870 uintptr_t thunk; member
|
| /external/swiftshader/third_party/llvm-subzero/include/llvm/Support/ |
| D | MachO.h | 866 uint32_t thunk; member
872 uint64_t thunk; member
878 uintptr_t thunk; member
|
| /external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/BinaryFormat/ |
| D | MachO.h | 899 uint32_t thunk; member
905 uint64_t thunk; member
911 uintptr_t thunk; member
|
| /external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/BinaryFormat/ |
| D | MachO.h | 930 uint32_t thunk; member
936 uint64_t thunk; member
942 uintptr_t thunk; member
|
| /external/google-breakpad/src/third_party/mac_headers/mach-o/ |
| D | loader.h | 1566 void* (*thunk)(struct tlv_descriptor*); member
|