Home
last modified time | relevance | path

Searched refs:output_count (Results 1 – 25 of 28) sorted by relevance

12

/external/mesa3d/src/compiler/nir/
Dnir_xfb_info.h58 uint16_t output_count; member
68 nir_xfb_info_size(uint16_t output_count) in nir_xfb_info_size() argument
70 return sizeof(nir_xfb_info) + sizeof(nir_xfb_output_info) * output_count; in nir_xfb_info_size()
Dnir_gather_xfb_info.c48 nir_xfb_info_create(void *mem_ctx, uint16_t output_count) in nir_xfb_info_create() argument
50 return rzalloc_size(mem_ctx, nir_xfb_info_size(output_count)); in nir_xfb_info_create()
145 nir_xfb_output_info *output = &xfb->outputs[xfb->output_count++]; in add_var_xfb_outputs()
266 qsort(xfb->outputs, xfb->output_count, sizeof(xfb->outputs[0]), in nir_gather_xfb_info_with_varyings()
278 for (unsigned i = 0; i < xfb->output_count; i++) { in nir_gather_xfb_info_with_varyings()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op.cc100 const int output_count = ctx->num_outputs(); in Compute() local
101 CHECK(output_count == execute_info_.graph_output_node_name_size() && in Compute()
102 output_count == output_types_.size()); in Compute()
103 for (int i = 0; i < output_count; ++i) { in Compute()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dunpack.cc32 const TfLiteEvalTensor* input, int output_count, in UnpackImpl() argument
62 for (int i = 0; i < output_count; ++i) { in UnpackImpl()
67 int loc = k * output_count * copy_size + i * copy_size; in UnpackImpl()
Dsplit.cc30 const int output_count = NumOutputs(node); in SplitImpl() local
42 int64_t split_size = output_dims->data[axis] * output_count; in SplitImpl()
57 for (int i = 0; i < output_count; ++i) { in SplitImpl()
Dsplit_v.cc41 const int output_count = NumOutputs(node); in SplitImpl() local
42 for (int i = 0; i < output_count; i++) { in SplitImpl()
59 for (int i = 0; i < output_count; ++i) { in SplitImpl()
/external/tensorflow/tensorflow/core/util/
Dguarded_philox_random.h66 random::PhiloxRandom ReserveRandomOutputs(int64 output_count, in ReserveRandomOutputs() argument
68 int64 conservative_sample_count = output_count * multiplier; in ReserveRandomOutputs()
/external/tensorflow/tensorflow/core/ops/
Dremote_fused_graph_ops_test.cc29 auto set_n = [&op](int input1_count, int input2_count, int output_count) { in TEST() argument
41 for (int i = 0; i < output_count; ++i) { in TEST()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dsoc_interface.h48 int output_count,
85 void* soc_interface_SetOneNodeOutputs(int output_count, int* max_size);
Dgraph_transferer_test.cc252 if (params.output_count() > 0) { in SanityCheckNodes()
256 EXPECT_EQ(params.output_count(), output_params->max_byte_size_size()); in SanityCheckNodes()
335 EXPECT_EQ(1, params_conv->output_count()); in TEST_F()
361 EXPECT_EQ(1, params_max_pool->output_count()); in TEST_F()
Dhexagon_control_wrapper.cc281 int output_count = 0; in SetupGraph() local
285 output_count = std::get<1>(output_ptr_and_count); in SetupGraph()
300 output_ptr, output_count); in SetupGraph()
Dgraph_transferer.cc1145 << (params.output_count() <= 0 in DumpNodeTransferParams()
1148 << ", size = " << params.output_count() << " ]"; in DumpNodeTransferParams()
1197 << (params.output_count() <= 0 in DumpVerificationStringOfNodeTransferParams()
1200 << "," << params.output_count() << "," << params.type_name() << "]"; in DumpVerificationStringOfNodeTransferParams()
/external/mesa3d/src/compiler/glsl/
Dgl_nir_link_xfb.c102 xfb_info->output_count); in gl_nir_link_assign_xfb_resources()
103 linked_xfb->NumOutputs = xfb_info->output_count; in gl_nir_link_assign_xfb_resources()
162 for (unsigned i = 0; i < xfb_info->output_count; i++) { in gl_nir_link_assign_xfb_resources()
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dtflite_engine.cc91 int output_count = OutputCount(interpreter); in GetOutputs() local
92 tensors.reserve(output_count); in GetOutputs()
93 for (int index = 0; index < output_count; index++) { in GetOutputs()
/external/tensorflow/tensorflow/lite/kernels/
Dunpack.cc89 const TfLiteTensor* input, int output_count, int axis) { in UnpackImpl() argument
92 op_params.num_split = output_count; in UnpackImpl()
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc185 int64 output_count = output_shape.num_elements(); in CalculateFlops() local
187 output_count * filter_in_depth * filter_height * filter_width * 2; in CalculateFlops()
198 int64 output_count = output_shape.num_elements(); in CalculateFlops() local
199 current_flops = k * output_count * 2; in CalculateFlops()
205 int64 output_count = output_shape.num_elements(); in CalculateFlops() local
206 current_flops = output_count * filter_height * filter_width * 2; in CalculateFlops()
/external/tensorflow/tensorflow/core/framework/
Dgraph_transfer_info.proto24 int32 output_count = 7; field
/external/mesa3d/src/freedreno/vulkan/
Dtu_shader.c696 for (int i = 0; i < xfb->output_count; i++) in tu_gather_xfb_info()
707 assert(xfb->output_count < IR3_MAX_SO_OUTPUTS); in tu_gather_xfb_info()
708 info->num_outputs = xfb->output_count; in tu_gather_xfb_info()
717 for (int i = 0; i < xfb->output_count; i++) { in tu_gather_xfb_info()
/external/mesa3d/src/amd/vulkan/
Dradv_shader_info.c517 assert(xfb->output_count < MAX_SO_OUTPUTS); in gather_xfb_info()
518 so->num_outputs = xfb->output_count; in gather_xfb_info()
520 for (unsigned i = 0; i < xfb->output_count; i++) { in gather_xfb_info()
/external/mesa3d/src/intel/vulkan/
Danv_pipeline_cache.c60 uint32_t xfb_info_size = nir_xfb_info_size(xfb_info_in->output_count); in anv_shader_bin_create()
119 xfb_info_in->output_count); in anv_shader_bin_create()
169 nir_xfb_info_size(shader->xfb_info->output_count); in anv_shader_bin_write_to_blob()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py4884 output_count = np.prod(output_shape.as_list(), dtype=np.int64)
4885 return ops.OpStats("flops", (output_count * filter_in_depth * filter_time *
4902 output_count = np.prod(output_shape.as_list(), dtype=np.int64)
4905 (output_count * filter_in_depth * filter_height * filter_width * 2))
4920 output_count = np.prod(output_shape.as_list(), dtype=np.int64)
4921 return ops.OpStats("flops", (output_count * filter_height * filter_width * 2))
5609 output_count = np.prod(output_shape.as_list(), dtype=np.int64)
5610 return ops.OpStats("flops", (output_count * filter_height * filter_width * 2))
Dmath_ops.py3582 output_count = np.prod(output_shape.as_list())
3583 return ops.OpStats("flops", (k * output_count * 2))
3599 output_count = np.prod(output_shape.as_list())
3600 return ops.OpStats("flops", (k * output_count * 2))
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dceva_tflm_lib.h516 int output_count, int copy_size);
519 int output_count, int copy_size);
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc89 const int output_count = output_names.size(); in TestTransformedVersusFloatGraph() local
90 EXPECT_EQ(output_count, float_outputs.size()); in TestTransformedVersusFloatGraph()
91 EXPECT_EQ(output_count, transformed_outputs.size()); in TestTransformedVersusFloatGraph()
92 for (int i = 0; i < output_count; ++i) { in TestTransformedVersusFloatGraph()
/external/mesa3d/src/gallium/drivers/r300/
Dr300_emit.c1113 unsigned output_count = MAX2(util_bitcount(code->OutputsWritten), 1); in r300_emit_vs_state() local
1117 vtx_mem_size / output_count, 10); in r300_emit_vs_state()

12