- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 115 for getBlob (0.15 sec)
-
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
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/tensorflow/transforms/collection_ops_util.cc
for (int64_t dim : element_shape) { buffer_shape.push_back(dim); } auto zero = CreateScalarConst(0, builder, op->getLoc()); if (getElementTypeOrSelf(zero.getType()) != element_dtype) { zero = builder.create<TF::CastOp>( op->getLoc(), ArrayRef<Type>{tensorflow::GetTypeFromTFTensorShape({}, element_dtype)}, ArrayRef<Value>{zero}); } auto buffer_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
// It is currently not possible to easily pack the output of a multi-result // op into an op with a single varidic output in `.td`. auto converted = rewriter.create<TFL::CustomOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), "TensorListPopBack", TFL::ConstBytesAttr::get(getContext(), "")); rewriter.replaceOp(op, converted.getResults()); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td
"$_builder.create<TF::ShapeOp>($0.getLoc(), $1, $2)">; def IsI32 : NativeCodeCall< "$_builder.getBoolAttr(getElementTypeOrSelf($0.getType()).isInteger(32))">; def CreateTFCastOpF32 : NativeCodeCall< "CreateTFCastOpF32(&$_builder, $0.getLoc(), $1, $2)">; def CreateTFCastOpI32 : NativeCodeCall< "CreateTFCastOpI32(&$_builder, $0.getLoc(), $1, $2)">; def CreateTensorScatterNdOp : NativeCodeCall<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
op.getLoc(), &rewriter); } else if (out_reshape_need) { out = createOutputReshapeOpForDynamic(out, reshape_shape, original_lhs, original_rhs, original_dnums, op.getLoc(), &rewriter); } out = createTransposeOp(out, op.getLoc(), out_transpose, &rewriter); rewriter.replaceOp(op, out);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
for (const auto& arg : func_op.getArguments()) { arg_locs.push_back(arg.getLoc()); } } // Creates a new main function. auto func_type = FunctionType::get(context, arg_types, result_types); auto main_func = builder.create<func::FuncOp>( module_op.getLoc(), kImportModelDefaultGraphFuncName, func_type); builder.createBlock(&main_func.getBody(), main_func.begin(), arg_types,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc
} } auto island = OpBuilder(fetch).create<IslandOp>( fetch.getLoc(), data_types, /*control=*/ControlType::get(fetch.getContext()), /*controlInputs=*/control_fetches); island.getBody().push_back(new Block); OpBuilder::atBlockEnd(&island.GetBody()) .create<YieldOp>(fetch.getLoc(), data_fetches); const int fetch_control_idx = data_fetches.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
return failure(); } auto tfl_fusable_op = builder.create<TFL::CustomOp>( func->getLoc(), func.getFunctionType().getResults(), func.getArguments(), custom_op_name, CustomOption(&builder, custom_option_buffer)); builder.create<func::ReturnOp>(func->getLoc(), tfl_fusable_op.getResults()); return success(); } // Abstracts the conversion of the embedded lookup composite function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
auto merged_execute_launch = builder->create<tf_device::LaunchOp>( merged_execute.getLoc(), execute_launch.getDeviceAttr(), merged_execute.getResultTypes()); merged_execute_launch.getBody().push_back(new Block); builder->setInsertionPointToEnd(&merged_execute_launch.GetBody()); builder->create<tf_device::ReturnOp>(merged_execute.getLoc(), merged_execute.getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0)