- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 380 for getLoc (0.13 sec)
-
tensorflow/compiler/mlir/tfr/passes/decompose.cc
builder.create<ConstOp>(op->getLoc(), output_type, attribute); } else { attr_cst = builder.create<mlir::arith::ConstantOp>( op->getLoc(), cast<TypedAttr>(attribute)); } new_operands.push_back(attr_cst); } } // Create the TFR call op auto new_op = builder.create<CallOp>( op->getLoc(), compose_func_type.getResults(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
builder.insert(var_handle_op_in_init); auto const_op = builder.create<mlir::arith::ConstantOp>( session_init_func.getLoc(), tensor_attr.getType(), tensor_attr); builder.create<TF::AssignVariableOp>( session_init_func.getLoc(), llvm::ArrayRef<mlir::Type>{}, llvm::ArrayRef<mlir::Value>{var_handle_op_in_init->getResult(0), const_op.getResult()}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
ArrayRef<Value> args; builder.create<func::CallOp>(session_initializer.getLoc(), func.getFunctionType().getResults(), func.getSymName(), args); } builder.create<func::ReturnOp>(session_initializer.getLoc()); session_initializer.erase(); } } // namespace std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
Type output_type = bmm_op.getResult().getType(); auto no_input = rewriter.create<TFL::NoValueOp>( bmm_op->getLoc(), rewriter.getNoneType(), rewriter.getUnitAttr()); auto fc_op = rewriter.create<TFL::FullyConnectedOp>( bmm_op->getLoc(), ArrayRef<Type>{output_type}, /*input=*/output_lhs, /*filter=*/output_rhs, /*bias=*/no_input,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
builder.create<mlir::TF::YieldOp>(if_region.getLoc(), /*operands=*/ArrayRef<Value>{}); // Create empty else branch region. auto& else_branch = host_side_if.getElseBranch(); else_branch.push_back(new Block); builder.setInsertionPointToEnd(&else_branch.front()); builder.create<mlir::TF::YieldOp>(if_region.getLoc(),
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/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
replacer.create<mlir::func::ReturnOp>(yield_op.getLoc(), yield_op.getOperands()); yield_op.erase(); } // Remap the captured operands in the (former) island block with newly // created entry block arguments in the function body. { Block &entry_block = outlined_func.getBody().front(); auto loc = outlined_func.getLoc(); for (Value operand : operands) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
} } else { bias = rewriter .create<AddOp>(add_op.getLoc(), bias, constant_val, rewriter.getStringAttr("NONE")) .getOutput(); } auto fc = rewriter.create<TFL::FullyConnectedOp>( FusedLoc::get(fc_op.getContext(), {fc_op.getLoc(), add_op.getLoc()}), add_op.getType(), /*input=*/fc_op.getInput(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
} operand = rewriter.create<mhlo::BitcastConvertOp>(op->getLoc(), *output_type, operand); Value res_min_clipped = rewriter.create<chlo::BroadcastMaxOp>( op->getLoc(), *output_type, operand, *min_or, broadcast_dims); Value res_max_clipped = rewriter.create<chlo::BroadcastMinOp>( op->getLoc(), *output_type, res_min_clipped, *max_or, broadcast_dims);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
Value scalar_zero = CreateI32SplatConst(op.getLoc(), rewriter, {}, 0); Type shape_dtype = getElementTypeOrSelf(op.getElementShape().getType()); Value num_elements = operands[1]; IntegerAttr attr; if (matchPattern(num_elements, m_Constant(&attr))) { return CreateI32SplatConst(op.getLoc(), rewriter, {1}, attr.getInt()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
} auto anonymous_iterator = builder.create<AnonymousIteratorV3Op>( reduce_dataset.getLoc(), RankedTensorType::get({}, builder.getType<ResourceType>()), /*output_types=*/builder.getArrayAttr(type_attrs), /*shape_types=*/builder.getArrayAttr(shape_attrs)); builder.create<MakeIteratorOp>(reduce_dataset.getLoc(), reduce_dataset.getInputDataset(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0)