- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 106 for created (0.64 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
builder, block.getParent()->getLoc(), execute, device); builder->create<tf_device::ReturnOp>(block.getParent()->getLoc(), block_launch_op.getResults()); } return success(); } // Creates a `tf.TPUCompileSucceededAssert` operation that parses compilation // status of `compile_op` to check whether compilation is successful.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
ModuleOp module) { SymbolTableCollection table; SymbolUserMap symbol_map(table, module); llvm::DenseMap<func::FuncOp, llvm::DenseSet<func::FuncOp>> caller_callee_map; // Creates work queue for determining reachability below. std::queue<func::FuncOp> function_worklist; for (auto func : module.getOps<func::FuncOp>()) { for (auto user : symbol_map.getUsers(func)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto reshape_sizes = DenseIntElementsAttr::get(reshape_type, new_shape); auto reshape_value = builder->create<TF::ConstOp>(input.getLoc(), reshape_sizes); return builder->create<TF::ReshapeOp>(input.getLoc(), reshape_result_type, input, reshape_value); } // Creates transpose op for shape to depth transform. TF::TransposeOp GetTransposeOpForConv2DFilter(OpBuilder* builder, Value input) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
} else { if (IsCompatibleTypeWithTFLCastOp(value.getType()) && IsCompatibleTypeWithTFLCastOp(type)) { auto cast = b.create<CastOp>(yield_op->getLoc(), type, value); args.push_back(cast); } else { auto cast = b.create<TF::CastOp>(yield_op->getLoc(), type, value); args.push_back(cast); } } } args.append(new_args.begin(), new_args.end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
} // Creates a wrapper island enclosing the `sub_op` dependent on // `control_inputs`. IslandOp CreateIsland(Operation* sub_op, ValueRange control_inputs, OpBuilder builder) { assert(sub_op); auto control_type = ControlType::get(builder.getContext()); auto island = builder.create<IslandOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
auto s_qconst = builder.create<SparseQConstOp>(op->getLoc(), cst.getQtypeAttr(), cst.getValue(), s_param, new_value); value.replaceAllUsesWith(s_qconst.getResult()); cst.erase(); } if (result.needs_densify) { auto value = op->getOperand(operand); auto densify = builder.create<DensifyOp>(op->getLoc(), value.getType(), value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
RankedTensorType::get(static_cast<int32_t>(1), builder.getI32Type()); auto out_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_out_segids)); auto contracting_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_contracting_segids)); auto num_segids_tensor = builder.create<TFL::ConstOp>(DenseIntElementsAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.cc
"Disable deadness related safety checks when clustering (this is " "unsound)."), Flag("tf_xla_disable_resource_variable_safety_checks_for_debugging", &mark_for_compilation_flags ->tf_xla_disable_resource_variable_safety_checks_for_debugging, "Disable resource variables related safety checks when clustering " "(this is unsound)."),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
return a->isBeforeInBlock(b); }); // Create tf_device::ClusterOp before the last operation in the block that // is a part of a match set. auto back = cluster.operations.back(); auto loc = back->getLoc(); OpBuilder builder(back); auto cluster_op = builder.create<tf_device::ClusterOp>(loc, return_types, policy); // Create block in cluster_op's region and move 'cluster.operations' into
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
Value tensor_i32 = builder.create<TF::CastOp>( loc, tensor_type.clone(builder.getIntegerType(32)), tensor); auto reduced = builder.create<TF::SumOp>(loc, tensor_i32, reduction_indices_value, /*keep_dims=*/builder.getBoolAttr(true)); auto mul_op = builder.create<TF::MulOp>(loc, zp, reduced);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0)