Home
last modified time | relevance | path

Searched defs:output_count (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dguarded_philox_random.h66 random::PhiloxRandom ReserveRandomOutputs(int64_t output_count, in ReserveRandomOutputs()
/external/mesa3d/src/compiler/nir/
Dnir_xfb_info.h63 uint16_t output_count; member
73 nir_xfb_info_size(uint16_t 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()
/external/pytorch/aten/src/ATen/nnapi/
Dnnapi_model_loader.cpp56 uint32_t output_count; member
65 int32_t output_count; member
/external/tensorflow/tensorflow/lite/kernels/
Dunpack.cc89 const TfLiteTensor* input, int output_count, int axis) { in UnpackImpl()
/external/tensorflow/tensorflow/core/framework/
Dgraph_transfer_info.proto24 int32 output_count = 7; field
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc189 int64_t output_count = output_shape.num_elements(); in CalculateFlops() local
202 int64_t output_count = output_shape.num_elements(); in CalculateFlops() local
209 int64_t output_count = output_shape.num_elements(); in CalculateFlops() local
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dtflite_engine.cc84 int output_count = OutputCount(interpreter); in GetOutputs() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_embedding_ops.cc97 int32 output_count = tpu_embedding_config_.feature_descriptor_size(); in Compile() local
/external/tensorflow/tensorflow/lite/c/
Dc_api_experimental.cc183 int32_t output_count = TfLiteSignatureRunnerGetOutputCount(signature_runner); in TfLiteSignatureRunnerGetOutputName() local
/external/deqp/external/vulkancts/modules/vulkan/conditional_rendering/
DvktConditionalTransformFeedbackTests.cpp298 const uint32_t output_count = 4 * 6; // 4 stream, 6 points in createXfbBuffer() local
/external/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_ml.h93 unsigned output_count; member
/external/mesa3d/src/gallium/frontends/teflon/
Dtfl_device.c56 unsigned output_count; member
/external/mesa3d/src/panfrost/util/
Dpan_ir.h287 unsigned output_count; member
/external/mesa3d/src/gallium/drivers/r300/
Dr300_emit.c1106 unsigned output_count = MAX2(util_bitcount(code->OutputsWritten), 1); in r300_emit_vs_state() local
/external/coreboot/src/soc/mediatek/mt8173/include/soc/
Dddp.h254 u32 output_count; member
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc89 const int output_count = output_names.size(); in TestTransformedVersusFloatGraph() local
/external/mesa3d/src/gallium/include/pipe/
Dp_state.h1088 unsigned output_count; member
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1124 uint32_t output_count, in AddOperationToModel()