Searched defs:num_stages (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpcbench_test.cc | 124 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, in CreateGraphDef()
169 static void BM_Helper(int iters, int width, int num_stages, int tensor_size, in BM_Helper()
215 static void BM_ShardedProgram(int iters, int width, int num_stages) { in BM_ShardedProgram()
240 static void BM_SingleDevice(int iters, int width, int num_stages) { in BM_SingleDevice()
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | trivial_test_graph_input_yielder.cc | 31 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, in CreateGraphDef()
90 int num_stages, int width, int tensor_size, bool insert_queue, in TrivialTestGraphInputYielder()
|
| /external/mesa3d/src/mesa/drivers/dri/i965/ |
| D | brw_context.c | 362 unsigned num_stages = 0; in brw_initialize_context_constants() local
|
| /external/mesa3d/src/intel/vulkan/ |
| D | genX_cmd_buffer.c | 1475 const unsigned num_stages = in cmd_buffer_alloc_push_constants() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1239 int64 num_stages = tensorflow::Log2Ceiling(dimension_to_sort_bound); in HandleSort() local
|