Home
last modified time | relevance | path

Searched refs:state_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsvdf.h71 const int32_t* bias_data, int16_t* state_data, in EvalIntegerSVDF() argument
86 std::copy(state_data + 1, state_data + n_batch * n_memory * n_filter, in EvalIntegerSVDF()
87 state_data); in EvalIntegerSVDF()
94 int16_t* result_in_batch = state_data + (n_memory - 1); in EvalIntegerSVDF()
121 const int16_t* state_data_batch = state_data + b * n_memory * n_filter; in EvalIntegerSVDF()
157 const float* bias_data, float* scratch_data, float* state_data, in EvalFloatSVDF() argument
169 std::copy(state_data + 1, state_data + batch_size * memory_size * num_filters, in EvalFloatSVDF()
170 state_data); in EvalFloatSVDF()
183 state_data[i * memory_size + memory_size - 1] = scratch_data[i]; in EvalFloatSVDF()
188 bias_data, params->activation, state_data, scratch_data, output_data); in EvalFloatSVDF()
/external/tensorflow/tensorflow/core/kernels/
Dstateful_random_ops_gpu.cu.cc41 StateElementType* __restrict__ state_data, in FillKernel() argument
48 *philox = GetPhiloxRandomFromMem(state_data); in FillKernel()
59 UpdateMemWithPhiloxRandom(*philox, output_size, state_data); in FillKernel()
81 auto state_data = state_tensor_flat.data(); in operator ()() local
98 d.stream(), dist, state_size, output_size, state_data, output_data)); in operator ()()
Dstateful_random_ops.cc42 auto state_data = state_tensor_flat.data(); in operator ()() local
44 auto philox = GetPhiloxRandomFromMem(state_data + alg_tag_skip); in operator ()()
45 UpdateMemWithPhiloxRandom(philox, output_size, state_data + alg_tag_skip); in operator ()()
/external/vulkan-validation-layers/layers/
Dcore_validation_types.h1188 …void initGraphicsPipeline(ValidationStateTracker *state_data, const VkGraphicsPipelineCreateInfo *…
1190 …void initComputePipeline(ValidationStateTracker *state_data, const VkComputePipelineCreateInfo *pC…
1191 …void initRayTracingPipelineNV(ValidationStateTracker *state_data, const VkRayTracingPipelineCreate…
Ddescriptor_sets.cpp549 … cvdescriptorset::DescriptorSet::StateTracker *state_data) in DescriptorSet() argument
554 state_data_(state_data), in DescriptorSet()
557 pool_state_ = state_data->GetDescriptorPoolState(pool); in DescriptorSet()
Dcore_validation.cpp723 …::DescriptorSetLayout const> const GetDescriptorSetLayout(const ValidationStateTracker *state_data, in GetDescriptorSetLayout() argument
725 auto it = state_data->descriptorSetLayoutMap.find(dsLayout); in GetDescriptorSetLayout()
726 if (it == state_data->descriptorSetLayoutMap.end()) { in GetDescriptorSetLayout()
8605 … const VkImageMemoryBarrier &barrier, const IMAGE_STATE *state_data) { in ValidateBarrierQueueFamilies() argument
8607 if (!state_data) { in ValidateBarrierQueueFamilies()
8612 barrier_queue_families::ValidatorState val(this, func_name, cb_state, &barrier, state_data); in ValidateBarrierQueueFamilies()
8620 … const VkBufferMemoryBarrier &barrier, const BUFFER_STATE *state_data) { in ValidateBarrierQueueFamilies() argument
8622 if (!state_data) { in ValidateBarrierQueueFamilies()
8627 barrier_queue_families::ValidatorState val(this, func_name, cb_state, &barrier, state_data); in ValidateBarrierQueueFamilies()
14503 void PIPELINE_STATE::initGraphicsPipeline(ValidationStateTracker *state_data, const VkGraphicsPipel… in initGraphicsPipeline() argument
[all …]
Dcore_validation.h973 const IMAGE_STATE* state_data);
975 const BUFFER_STATE* state_data);