Searched refs:state_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops_gpu.cu.cc | 33 StateElementType* state_data, in FillKernel() argument
40 *philox = GetPhiloxRandomFromMem(state_data); in FillKernel()
50 UpdateMemWithPhiloxRandom(*philox, output_size, state_data); in FillKernel()
67 auto state_data = state_tensor_flat.data(); in operator ()() local
80 state_data, output_data)); in operator ()()
|
| D | stateful_random_ops.cc | 32 auto state_data = state_tensor_flat.data(); in operator ()() local
34 auto philox = GetPhiloxRandomFromMem(state_data + alg_tag_skip); in operator ()()
35 UpdateMemWithPhiloxRandom(philox, output_size, state_data + alg_tag_skip); in operator ()()
|
| /external/vulkan-validation-layers/layers/ |
| D | core_validation.cpp | 8249 … const VkImageMemoryBarrier *barrier, const IMAGE_STATE *state_data) { in ValidateBarrierQueueFamilies() argument
8251 if (!state_data) { in ValidateBarrierQueueFamilies()
8256 … barrier_queue_families::ValidatorState val(device_data, func_name, cb_state, barrier, state_data); in ValidateBarrierQueueFamilies()
8264 … const VkBufferMemoryBarrier *barrier, const BUFFER_STATE *state_data) { in ValidateBarrierQueueFamilies() argument
8266 if (!state_data) { in ValidateBarrierQueueFamilies()
8271 … barrier_queue_families::ValidatorState val(device_data, func_name, cb_state, barrier, state_data); in ValidateBarrierQueueFamilies()
|