- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 175 for pushBack (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// `replicate`. llvm::SmallVector<Value, 8> reformat_operands; for (const auto& entry : execute_arg_to_outer_args) { reformat_operands.push_back(execute.getArgs()[entry.first]); } reformat_operands.push_back(compile_launch.getResult(1)); reformat_operands.push_back(replicate.GetBody().getArgument( replicate.GetNumReplicatedBlockArguments() - 1)); builder.setInsertionPoint(execute_launch);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
for (int i = 0; i < noutputs; ++i) { output_names_vec.push_back(string(output_names[i])); } } // Process control output names. std::vector<string> control_output_names_vec; if (control_output_names) { control_output_names_vec.reserve(ncontrol_outputs); for (int i = 0; i < ncontrol_outputs; ++i) { control_output_names_vec.push_back(string(control_output_names[i])); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
if (is_guaranteed_constant(*n)) { return errors::InvalidArgument( "Guaranteed constants are not supported (", n->name(), ")"); } args.push_back(n); } else if (n->type_string() == "_Retval") { retvals.push_back(n); } } if (std::find(args.begin(), args.end(), nullptr) != args.end()) { return errors::InvalidArgument("Missing or non-consecutive arguments"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
for (int i = 0; i < repeats; ++i) state_->types.push_back(UnrankedTensorType::get(type_attr.getType())); } else if (output_arg.type() != tensorflow::DT_INVALID) { for (int i = 0; i < repeats; ++i) { Type type; TF_RETURN_IF_ERROR( ConvertDataType(output_arg.type(), builder, &type)); state_->types.push_back(type); } } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.cc
CHECK_EQ(arg.name(), api_def_arg.name()); bool is_list = ArgIsList(arg); output_types.push_back( strings::StrCat("::tensorflow::", is_list ? "OutputList" : "Output")); output_names.push_back(AvoidCPPKeywords(api_def_arg.rename_to())); is_list_output.push_back(is_list); } strings::StrAppend(&comment, "\nReturns:\n"); if (graph_op_def.output_arg_size() == 0) { // No outputs.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
outputs_replaced.push_back(user.getResult()); output_types.push_back(user.getType()); } else if (isa<IntegerType>(result_element_type)) { // If the result is an integer tensor, then it doesn't require the // dequantize op in the pattern. outputs_replaced.push_back(result); output_types.push_back(result.getType()); } else { return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
if (rhs_batch_idx.contains(i)) { if (rhs_only_batch) { rhs_eq.push_back(c); lhs_batch_dims.push_back(c); c++; } else { rhs_eq.push_back(lhs_batch_dims[batch_trace_idx]); batch_trace_idx++; } } else if (rhs_contract_idx.contains(i)) { absl::StrAppend(&rhs_eq, std::string(1, lhs_contract_dims[contract_trace_idx]));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
rmax = std::max(std::abs(rmin), std::abs(rmax)); rmin = -rmax; } TensorRangeSanityCheck(op, rmin, rmax); mins.push_back(rmin); maxs.push_back(rmax); } quant_type = quantfork::fakeQuantAttrsToType( op.getLoc(), num_bits, *op.getAxis(), mins, maxs, narrow_range, expressed, is_signed);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
op->getLoc(), unconstrainted_tensor_type, op->getOperand(i)); variadic_operands.push_back(casted); } auto build_list_op = builder.create<BuildListOp>( op->getLoc(), list_type, variadic_operands); new_operands.push_back(build_list_op.getOut()); } else { auto attr_name = compose_func.getArgAttrOfType<StringAttr>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
result_types.push_back(op->getOpResult(2).getType()); // Populate attributes. SmallVector<NamedAttribute, 4> attributes; // Activation will always be tanh. attributes.push_back(rewriter.getNamedAttr("fused_activation_function", rewriter.getStringAttr("TANH"))); // cell_clip. attributes.push_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0)