- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 194 for pushBack (0.16 sec)
-
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/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/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) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
if (!mlir::isa<BlockArgument>(value)) island_control_operands.push_back(value); } else { inner_op_operands.push_back(value); } } auto island = rewriter.create<tf_executor::IslandOp>( loc, new_types, island_control_operands); island.getBody().push_back(new mlir::Block); rewriter.setInsertionPointToEnd(&island.getBody().front());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/hardwares/target_hardware.cc
for (auto& device_spec : specified_device_specs) { auto device = GetCanonicalHardwareName(device_spec); if (device == "CPU") cpu_include = true; device_specs->push_back(device); } if (!cpu_include) { device_specs->push_back("CPU"); } // Make sure all the devices are registered. for (const std::string& device : *device_specs) { if (GetTargetHardware(device) == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 09 21:39:59 UTC 2023 - 9.9K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
// GradientTape which uses ones. Gradient* zero = input_tensors[target_index].ZerosLike(); new_zeros.push_back(zero); in_grads.push_back(zero); } else { in_grads.push_back(nullptr); } } else { in_grads.push_back(current_grad->second); } } // Avoid infinite recursion. Whichever forward function we run, it'll end up
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 12:40:29 UTC 2024 - 47.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
for (auto list_input : list_op.getOperands()) { auto cast_op = dyn_cast_or_null<CastOp>(list_input.getDefiningOp()); if (!cast_op) return failure(); list_inputs.push_back(cast_op.getArg()); list_input_types.push_back(cast_op.getInputElementType()); } CastValuesToSameType(rewriter, call_op.getLoc(), list_input_types, list_inputs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
auto dequantize_op = llvm::cast<DequantizeOp>(returned_op); Value dequantized_result = dequantize_op.getInput(); output_types.push_back(dequantized_result.getType()); terminator->setOperand(i, dequantized_result); returned_op->erase(); } else { output_types.push_back(returned_value.getType()); } } auto new_func_type = builder.getFunctionType(input_types, output_types);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc
std::vector<TFE_TensorHandle*> compute_fn_inputs; TFE_TensorHandle* input_a = TestScalarTensorHandle(ctx, 2.0f); TFE_TensorHandle* input_b = TestScalarTensorHandle(ctx, 1.0f); compute_fn_inputs.push_back(input_a); compute_fn_inputs.push_back(input_b); TFE_Op* compute_fn_op = TF_ConcreteFunctionMakeCallOp( compute_fn, compute_fn_inputs.data(), compute_fn_inputs.size(), status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 08:08:45 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
launch_results.push_back(result); launch_result_types.push_back(result.getType()); } } } before ? builder->setInsertionPoint(op) : builder->setInsertionPointAfter(op); auto launch = builder->create<mlir::tf_device::LaunchOp>( op->getLoc(), builder->getStringAttr(host_device), launch_result_types); launch.getBody().push_back(launch_block);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0)