- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 248 for getUses (0.44 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// to it have the same shape, we need to follow use-def chain of ops that // (conceptually) modify it i.e., ops that take an input TensorList and // produce an output TensorList. for (auto& use : tensorlist.getUses()) { if (auto push = llvm::dyn_cast<TensorListPushBackOp>(use.getOwner())) { auto element_type = mlir::dyn_cast<RankedTensorType>(push.getTensor().getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
// with corresponding results of new parallel_execute op for (auto& use : llvm::make_early_inc_range( old_parallel_execute->getResult(output_index).getUses())) { use.set(new_parallel_execute->getResult(output_index)); } continue; } int tpu_cluster_output_index = output_index - num_results_pre_cluster; const auto& output_sharding =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
has_dynamic_output = true; for (Operation* user : v.getUsers()) { if (!HasOutsideCompilationAncestor(user) && !llvm::isa<mlir::tf_device::ReturnOp>(user)) return true; } } if (!tensorflow::TypeValidForXLA(v.getType())) for (const Operation* user : v.getUsers()) if (!llvm::isa<mlir::tf_device::ReturnOp>(user))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
llvm::SmallVector<Value, 4> next_values_to_visit; for (Value value_to_visit : values_to_visit) { if (!visited_values.insert(value_to_visit).second) continue; for (auto& use : value_to_visit.getUses()) { Operation* owner = use.getOwner(); if (auto sharding = llvm::dyn_cast<mlir::TF::XlaShardingOp>(owner)) return sharding.get_XlaSharding();
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
// Only `mlir::func::CallOp` is supported as this requires knowing how // to rewrite arguments and results to a function. if (!isa<mlir::func::CallOp>(use.getUser())) continue; auto caller_parent_func = use.getUser()->getParentOfType<func::FuncOp>(); if (!caller_parent_func) continue; FuncToRewrite func_to_rewrite{/*original=*/caller_parent_func,
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/resource_op_lifting.cc
ResourceInfo info; info.data_type = GetResourceSubtype(resource); llvm::BitVector written_regions(op_->getNumRegions()); bool unsupported_use = false; for (OpOperand& use : resource.getUses()) { Operation* user = use.getOwner(); // If the user is not in one of the regions, we are not interested in it. // Since all the sub-regions within this region (i.e., regions attached to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
for (auto& use : uses_optional.value()) { if (llvm::isa<TF::WhileOp>(use.getUser())) { if (while_op != nullptr) { use.getUser()->emitOpError() << "multiple users of function."; return LogicalResult::failure(); } else { while_op = llvm::cast<TF::WhileOp>(use.getUser()); } } else { use.getUser()->emitOpError() << "non while use of function."; return LogicalResult::failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
!(qtype.isa<mlir::quant::UniformQuantizedType>() || qtype.isa<mlir::quant::UniformQuantizedPerAxisType>())) { return; } for (auto& use : value.getUses()) { Operation* user = use.getOwner(); if (user->hasTrait<mlir::OpTrait::IsTerminator>()) continue; auto affine_user = llvm::dyn_cast<mlir::AffineQuantizedOpInterface>(user); if (affine_user &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
// marked as a stateful. If so, set the tensor's is_variable as true // This is v1 ref variable semantics in the TFLite runtime. bool is_variable = false; for (auto& use : value.getUses()) { is_variable = IsStatefulOperand(use.getOwner(), use.getOperandNumber()); if (is_variable) { break; } } // The value is used as a variable if produced by an op with "tfl.is_variable"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tests/benchmark_test.go
func BenchmarkCreate(b *testing.B) { user := *GetUser("bench", Config{}) for x := 0; x < b.N; x++ { user.ID = 0 DB.Create(&user) } } func BenchmarkFind(b *testing.B) { user := *GetUser("find", Config{}) DB.Create(&user) for x := 0; x < b.N; x++ { DB.Find(&User{}, "id = ?", user.ID) } } func BenchmarkScan(b *testing.B) { user := *GetUser("scan", Config{}) DB.Create(&user)
Registered: Wed Jun 12 16:27:09 UTC 2024 - Last Modified: Wed Jun 01 03:50:57 UTC 2022 - 1.5K bytes - Viewed (0)