Home
last modified time | relevance | path

Searched refs:num_stages (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc124 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, in CreateGraphDef() argument
138 for (int i = 0; i < num_stages; i++) { in CreateGraphDef()
169 static void BM_Helper(int iters, int width, int num_stages, int tensor_size, in BM_Helper() argument
176 GraphDef def = CreateGraphDef(num_stages, width, tensor_size, in BM_Helper()
215 static void BM_ShardedProgram(int iters, int width, int num_stages) { in BM_ShardedProgram() argument
216 BM_Helper(iters, width, num_stages, 2 /*tensor_size*/, true /*multi-device*/); in BM_ShardedProgram()
240 static void BM_SingleDevice(int iters, int width, int num_stages) { in BM_SingleDevice() argument
241 BM_Helper(iters, width, num_stages, 2 /*tensor_size*/, in BM_SingleDevice()
/external/tensorflow/tensorflow/core/grappler/inputs/
Dtrivial_test_graph_input_yielder.cc31 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, in CreateGraphDef() argument
46 for (int i = 0; i < num_stages; i++) { in CreateGraphDef()
90 int num_stages, int width, int tensor_size, bool insert_queue, in TrivialTestGraphInputYielder() argument
92 : num_stages_(num_stages), in TrivialTestGraphInputYielder()
Dtrivial_test_graph_input_yielder.h31 TrivialTestGraphInputYielder(int num_stages, int width, int tensor_size,
/external/mesa3d/src/mesa/drivers/dri/i965/
Dbrw_context.c362 unsigned num_stages = 0; in brw_initialize_context_constants() local
365 num_stages++; in brw_initialize_context_constants()
446 ctx->Const.MaxUniformBufferBindings = num_stages * BRW_MAX_UBO; in brw_initialize_context_constants()
447 ctx->Const.MaxCombinedUniformBlocks = num_stages * BRW_MAX_UBO; in brw_initialize_context_constants()
448 ctx->Const.MaxCombinedAtomicBuffers = num_stages * BRW_MAX_ABO; in brw_initialize_context_constants()
449 ctx->Const.MaxCombinedShaderStorageBlocks = num_stages * BRW_MAX_SSBO; in brw_initialize_context_constants()
450 ctx->Const.MaxShaderStorageBufferBindings = num_stages * BRW_MAX_SSBO; in brw_initialize_context_constants()
451 ctx->Const.MaxCombinedTextureImageUnits = num_stages * max_samplers; in brw_initialize_context_constants()
452 ctx->Const.MaxCombinedImageUniforms = num_stages * BRW_MAX_IMAGES; in brw_initialize_context_constants()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1239 int64 num_stages = tensorflow::Log2Ceiling(dimension_to_sort_bound); in HandleSort() local
1240 CHECK_GE(1ULL << num_stages, dimension_to_sort_bound); in HandleSort()
1241 CHECK_LT(1ULL << (num_stages - 1), dimension_to_sort_bound); in HandleSort()
1267 const uint64 kTileSize = std::min(2048ULL, 1ULL << num_stages); in HandleSort()
1275 uint64 standard_num_iterations_in_sort_dim = 1ULL << (num_stages - 1); in HandleSort()
1345 for (int64 stage = 0; stage < num_stages; ++stage) { in HandleSort()
/external/mesa3d/src/intel/vulkan/
DgenX_cmd_buffer.c1475 const unsigned num_stages = in cmd_buffer_alloc_push_constants() local
1477 unsigned size_per_stage = push_constant_kb / num_stages; in cmd_buffer_alloc_push_constants()