| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_op.cc | 76 const int input_count = ctx->num_inputs(); in Compute() local
78 CHECK(input_count == graph_input_count && in Compute()
79 input_count == input_types_.size()) in Compute()
80 << "input_count = " << input_count in Compute()
|
| D | meta_support.h | 99 int input_count, const quint8* bias, int bias_count,
106 void Clamp(OpKernelContext* context, const quint8* input, int input_count,
|
| D | meta_support.cc | 347 int input_count, const quint8* bias, int bias_count, in QuantizedBiasAdd() argument
362 params.kernel.rows = input_count / bias_count; in QuantizedBiasAdd()
|
| D | remote_fused_graph_execute_utils.cc | 773 int input_count = 0; in ClusterizeNodes() local
789 ++input_count; in ClusterizeNodes()
793 CHECK(input_count == 0 || input_count == node_in_edges_size) in ClusterizeNodes()
794 << "Invalid input_count(" << input_count << ", " in ClusterizeNodes()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | soc_interface.h | 47 bool soc_interface_AllocateInOutNodeBuffers(int input_count, int* input_sizes,
81 void* soc_interface_SetOneNodeInputs(int input_count, const int* const node_id,
|
| D | graph_transferer_test.cc | 241 if (params.input_count() > 0) { in SanityCheckNodes()
245 EXPECT_EQ(params.input_count(), input_params->node_input_size()); in SanityCheckNodes()
334 EXPECT_EQ(3, params_conv->input_count()); in TEST_F()
360 EXPECT_EQ(3, params_max_pool->input_count()); in TEST_F()
|
| D | hexagon_control_wrapper.cc | 279 const int input_count = std::get<1>(input_ptr_and_count); in SetupGraph() local
299 op_id, padding_id, input_ptr, input_count, in SetupGraph()
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | identity.py | 44 input_count = (2 if parameters["op_to_use"] == "identity_n_with_2_inputs"
49 for _ in range(input_count)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | immutable_executor_state.h | 47 input_count(0), in FrameInfo()
57 int input_count; member
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| D | tflite_engine.cc | 80 int input_count = InputCount(interpreter); in GetInputs() local
81 tensors.reserve(input_count); in GetInputs()
82 for (int index = 0; index < input_count; index++) { in GetInputs()
|
| /external/brotli/c/tools/ |
| D | brotli.c | 106 size_t input_count; member
183 size_t input_count = 0; in ParseParams() local
216 input_count++; in ParseParams()
515 params->input_count = input_count; in ParseParams()
520 if (input_count > 1 && output_set) return COMMAND_INVALID; in ParseParams()
689 if (context->input_count == 0) { in NextFile()
1049 context.input_count = 0; in main()
|
| /external/webrtc/modules/desktop_capture/ |
| D | desktop_region_unittest.cc | 293 int input_count; in TEST() member
340 r.AddRects(cases[i].input_rects, cases[i].input_count); in TEST()
345 for (int j = cases[i].input_count - 1; j >= 0; --j) { in TEST()
432 int input_count; in TEST() member
456 DesktopRegion r(cases[i].input_rects, cases[i].input_count); in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_transfer_info.proto | 23 int32 input_count = 6; field
|
| /external/mesa3d/src/gallium/frontends/lavapipe/ |
| D | lvp_pass.c | 116 for (uint32_t j = 0; j < subpass->input_count; ++j) { in lvp_render_pass_compile()
212 subpass->input_count = desc->inputAttachmentCount; in lvp_CreateRenderPass()
|
| /external/crosvm/devices/src/virtio/video/encoder/ |
| D | libvda_encoder.rs | 336 input_count, in read_event()
341 input_count, in read_event()
|
| D | encoder.rs | 46 input_count: u32,
|
| D | mod.rs | 228 input_count: u32, in require_input_buffers()
236 self.src_params.min_buffers = input_count; in require_input_buffers()
1331 input_count, in process_event()
1336 input_count, in process_event()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unique_op.cc | 153 int64 input_count = input_shape.dimensions(0); in DataOutputFastPath() local
170 auto shifted = xla::SliceInDim(sorted_data, 0, input_count - 1, in DataOutputFastPath()
|
| /external/mesa3d/src/broadcom/vulkan/ |
| D | v3dv_pass.c | 78 for (uint32_t j = 0; j < subpass->input_count; j++) { in pass_find_subpass_range_for_attachments()
159 subpass->input_count = desc->inputAttachmentCount; in v3dv_CreateRenderPass()
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_pass.c | 274 for (uint32_t j = 0; j < subpass->input_count; ++j) { in radv_render_pass_compile()
394 subpass->input_count = desc->inputAttachmentCount; in radv_CreateRenderPass()
577 subpass->input_count = desc->inputAttachmentCount; in radv_CreateRenderPass2()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 132 Status ReorderOutputEdges(Graph* g, Node* n, int input_count, in ReorderOutputEdges() argument
149 if (new_src_output < input_count - resource_input_count) { in ReorderOutputEdges()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass.cc | 302 int input_count = absl::c_count_if( in ReplaceFunctionCallWithPartitionedCall() local
305 std::vector<Output> args(input_count); in ReplaceFunctionCallWithPartitionedCall()
|
| /external/mesa3d/src/intel/vulkan/ |
| D | anv_pass.c | 297 subpass->input_count = desc->inputAttachmentCount; in anv_CreateRenderPass()
538 subpass->input_count = desc->inputAttachmentCount; in anv_CreateRenderPass2()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 600 int64 input_count() const { return operand_count() / 2; } in input_count() function
604 return absl::MakeSpan(operands()).subspan(0, input_count()); in inputs()
609 return absl::MakeSpan(operands()).subspan(input_count(), operand_count()); in init_values()
1308 int64 input_count() const { return operand_count() / 2; } in input_count() function
1311 return absl::MakeSpan(operands()).subspan(0, input_count()); in input_arrays()
1315 return absl::MakeSpan(operands()).subspan(input_count(), operand_count()); in init_values()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2702 int64 input_count = 1; in HandleReduce() local
2705 input_count = hlo->shape().tuple_shapes_size(); in HandleReduce()
2706 CHECK_GT(input_count, 0); in HandleReduce()
2719 for (int64 operand_id = 0; operand_id < input_count; ++operand_id) { in HandleReduce()
2720 inits.push_back(GetPartitionedHlo(hlo->operand(operand_id + input_count)) in HandleReduce()
2735 std::vector<Shape*> new_operand_shapes(input_count * 2); in HandleReduce()
2736 for (int64 i = 0; i < input_count; ++i) { in HandleReduce()
2738 new_operand_shapes[i + input_count] = inits[i]->mutable_shape(); in HandleReduce()
2746 std::vector<HloInstruction*> input_hlos(input_count); in HandleReduce()
2747 for (int64 i = 0; i < input_count; ++i) { in HandleReduce()
[all …]
|