- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 340 for pushBack (0.24 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
sharding_for_args.push_back(operand_sharding); continue; } if (infer_from_computation) { auto arg_sharding = GetXlaShardingFromArg(arg, logical_device_vec); if (arg_sharding) { sharding_for_args.push_back(arg_sharding.value()); continue; } } sharding_for_args.push_back(std::nullopt); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
builder.setInsertionPoint(call); auto new_operands = llvm::to_vector(call.getArgOperands()); new_operands.push_back(token); auto new_result_types = llvm::to_vector(call.getResultTypes()); new_result_types.push_back(token.getType()); auto new_call = builder.create<func::CallOp>( call.getLoc(), new_result_types, new_symbol ? *new_symbol : call.getCallee(), new_operands);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
// Add new control input if its defining op is not already a defining // op for some other operand. Update defining_ops. if (defining_ops.insert(new_control_input.getDefiningOp()).second) { operands.push_back(new_control_input); } } state.addOperands(operands); Operation* new_op = builder.create(state); item.replaceAllUsesWith(new_op); new_op->setAttrs(item.getAttrDictionary());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc
if (!main_fn) return std::string(""); flatbuffers::FlatBufferBuilder fb_builder; std::vector<mlir::func::FuncOp> funcs; funcs.push_back(main_fn); module.walk([&](mlir::func::FuncOp fn) { if (fn != main_fn) { funcs.push_back(fn); } }); // Populate the hardware metadata. // And collect the hardwares used. std::map<std::string, uint8_t> hardware_map;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 06:11:34 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/cc/client/client_session.cc
output_tensor_names.reserve(fetch_outputs.size()); for (auto const& output : fetch_outputs) { output_tensor_names.push_back(output.name()); } std::vector<string> target_node_names; target_node_names.reserve(run_outputs.size()); for (auto const& output : run_outputs) { target_node_names.push_back(output.node()->name()); } TF_RETURN_IF_ERROR(impl()->MaybeExtendGraph());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 09:04:10 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/cc/framework/gradients.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
func_operand_types.reserve(operands.size()); for (Value operand : operands) func_operand_types.push_back(operand.getType()); // Function results are the yield operands SmallVector<Type, 16> func_result_types; for (Value operand : island_op.GetYield().getOperands()) func_result_types.push_back(operand.getType()); FunctionType func_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc
// contain zero scale for zero values. llvm::SmallVector<double> scales; for (float scale : quant_params.scale) { if (scale == 0) { scales.push_back(std::numeric_limits<float>::min()); continue; } scales.push_back(scale); } // Scale size can't be zero as it is checked before. if (quant_params.scale.size() != 1) { return mlir::quant::UniformQuantizedPerAxisType::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
padding = "SAME"; } else { padding = "EXPLICIT"; explicit_padding.push_back(0); explicit_padding.push_back(0); explicit_padding.append(padding_array); explicit_padding.push_back(0); explicit_padding.push_back(0); } } CreateConvOp(conv_op, strides, padding, explicit_padding, dilation,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
Type result_type; // Include info about the activation function if applicable. if (fuse_activation) { locations.push_back(activation->getLoc()); fused_ops.push_back( StringAttr::get(context, activation->getName().stripDialect())); result_type = activation->getResultTypes().front(); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0)