Searched defs:num_stages (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/inputs/ |
H A D | trivial_test_graph_input_yielder.cc | 31 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 D | rpcbench_test.cc | 124 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 D | brw_context.c | 513 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 D | genX_cmd_buffer.c | 1032 const unsigned num_stages = local 1034 unsigned size_per_stage = push_constant_kb / num_stages;
|
Completed in 163 milliseconds