- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 107 for pushBack (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
SmallVector<int64_t, 8> perm0_values; for (const auto &value : perm0.getValues<APInt>()) perm0_values.push_back(value.getSExtValue()); SmallVector<int64_t, 8> perm1_values; for (const auto &value : perm1.getValues<APInt>()) perm1_values.push_back(value.getSExtValue()); for (int i = 0; i < perm0_values.size(); ++i) { if (perm0_values[perm1_values[i]] != i) return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
const std::string gpu1 = "/job:worker/replica:1/task:2/device:GPU:1"; llvm::SmallVector<std::unique_ptr<Device>, 2> devices; devices.push_back(FakeDevice::Make(cpu0)); devices.push_back(FakeDevice::Make(gpu0, "compute capability: 7.0")); devices.push_back(FakeDevice::Make(gpu1)); DeviceSet device_set; for (auto& device : devices) device_set.AddDevice(device.get()); AddDevicesToOp(*module_ref, &device_set);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
mlir::TensorType mlir_ty = mlir::UnrankedTensorType::get(elt_ty); input_tys.push_back(mlir_ty); } for (auto ty : output_dtys) { mlir::Type elt_ty; TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty)); mlir::TensorType mlir_ty = mlir::UnrankedTensorType::get(elt_ty); output_tys.push_back(mlir_ty); } llvm::SmallVector<mlir::NamedAttribute, 4> attrs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/cc/saved_model/loader_util.cc
// collection, so read from metagraph first. if (meta_graph_def.asset_file_def_size() > 0) { for (const auto& asset : meta_graph_def.asset_file_def()) { asset_file_defs->push_back(asset); } return absl::OkStatus(); } // Fall back to read from collection to be backward compatible with v1. const auto& collection_def_map = meta_graph_def.collection_def();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 10 10:25:28 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_filter.cc
if (!func_regex.match(func.getName())) { continue; } ApplyFunctionTacFilter(func, tac_filter.function_filter().filter_type(), builder); filtered_ops.push_back(func); } return; } llvm::Regex op_regex(tac_filter.op_filter().op_name_pattern()); module.walk([&](Operation* op) { auto named_loc = mlir::dyn_cast<NameLoc>(op->getLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
// Read all resource variables from the session. std::vector<std::string> variable_names; variable_names.reserve(resource_names.size()); for (StringRef name : resource_names) variable_names.push_back(name.str()); std::vector<Tensor> resource_tensors; Status status = session->Run( /*inputs=*/{}, variable_names, /*target_tensor_names=*/{}, &resource_tensors); if (!status.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
builder.setInsertionPointAfter(*old_launch_op); llvm::SmallVector<Type, 4> new_launch_op_results_types; for (Value result : new_launch_op_results) new_launch_op_results_types.push_back(result.getType()); auto new_launch_op = builder.create<tf_device::LaunchOp>( old_launch_op->getLoc(), old_launch_op->getDeviceAttr(), /*result_types=*/new_launch_op_results_types);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/jit/device_util_test.cc
device_info_cache.GetIdFor(device_name)); device_set.Insert(device_id); expected_devices.push_back(device_name); } device_set.ForEach([&](jit::DeviceId device_id) { actual_devices.push_back(string(device_info_cache.GetNameFor(device_id))); return true; }); EXPECT_EQ(expected_devices, actual_devices); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util_test.cc
cluster_func_ops.push_back(cluster_func); }); EXPECT_EQ(cluster_func_ops.size(), 1); EXPECT_TRUE(mlir::succeeded(tensorflow::EraseClusterFuncs(cluster_func_ops))); llvm::SmallVector<mlir::tf_device::ClusterFuncOp, 4> new_cluster_func_ops; module->walk([&](mlir::tf_device::ClusterFuncOp cluster_func) { new_cluster_func_ops.push_back(cluster_func); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cpu_device.cc
VLOG(1) << "Not creating XLA devices, tf_xla_enable_xla_devices not set " "and XLA device creation not requested"; return absl::OkStatus(); } devices->push_back(absl::StrCat("/physical_device:", DEVICE_XLA_CPU, ":0")); return absl::OkStatus(); } Status XlaCpuDeviceFactory::CreateDevices( const SessionOptions& session_options, const string& name_prefix,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.5K bytes - Viewed (0)