Searched defs:num_stages (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/inputs/
H A Dtrivial_test_graph_input_yielder.cc31 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, argument
46 for (int i = 0; i < num_stages; i++) {
90 int num_stages, int width, int tensor_size, bool insert_queue,
92 : num_stages_(num_stages),
/external/tensorflow/tensorflow/core/distributed_runtime/
H A Drpcbench_test.cc124 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, argument
138 for (int i = 0; i < num_stages; i++) {
169 static void BM_Helper(int iters, int width, int num_stages, int tensor_size, argument
176 GraphDef def = CreateGraphDef(num_stages, width, tensor_size,
215 static void BM_ShardedProgram(int iters, int width, int num_stages) { argument
216 BM_Helper(iters, width, num_stages, 2 /*tensor_size*/, true /*multi-device*/);
236 BM_Helper(iters, width, 2 /*num_stages*/, tensor_size, true /*multi-device*/);
240 static void BM_SingleDevice(int iters, int width, int num_stages) { argument
241 BM_Helper(iters, width, num_stages, 2 /*tensor_size*/,
/external/mesa3d/src/mesa/drivers/dri/i965/
H A Dbrw_context.c513 unsigned num_stages = 0; local
516 num_stages++;
582 ctx->Const.MaxUniformBufferBindings = num_stages * BRW_MAX_UBO;
583 ctx->Const.MaxCombinedUniformBlocks = num_stages * BRW_MAX_UBO;
584 ctx->Const.MaxCombinedAtomicBuffers = num_stages * BRW_MAX_ABO;
585 ctx->Const.MaxCombinedShaderStorageBlocks = num_stages * BRW_MAX_SSBO;
586 ctx->Const.MaxShaderStorageBufferBindings = num_stages * BRW_MAX_SSBO;
587 ctx->Const.MaxCombinedTextureImageUnits = num_stages * max_samplers;
588 ctx->Const.MaxCombinedImageUniforms = num_stages * BRW_MAX_IMAGES;
/external/mesa3d/src/intel/vulkan/
H A DgenX_cmd_buffer.c1032 const unsigned num_stages = local
1034 unsigned size_per_stage = push_constant_kb / num_stages;

Completed in 163 milliseconds