- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for addOperands (0.1 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
auto attr = ReadAttr(flex_buffer_map, &builder, custom_op.getCustomCode().str()); OperationState op_state(custom_op.getLoc(), custom_op.getCustomCode().str()); op_state.addOperands(custom_op.getOperands()); llvm::SmallVector<mlir::Type, 4> output_tys; for (int i = 0; i < custom_op.getNumResults(); i++) { output_tys.push_back(custom_op.getType(i)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
const std::string tf_op_full_name = llvm::Twine("tf.", tf_op_name).str(); // Create the TF op OperationState op_state(op.getLoc(), tf_op_full_name); op_state.addOperands(op.getOperands()); op_state.addTypes(op.getResultTypes()); SmallVector<NamedAttribute, 2> attrs; std::string parsed_op_name; tensorflow::NodeDef node_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
DCHECK(succeeded( VerifyCompatibleTypes(input.getType(), replicated_input.second))); state->addOperands(input); } block.addArgument(replicated_input.second, state->location); } for (auto packed_input : packed_inputs) { state->addOperands(packed_input); block.addArgument(packed_input.getType(), state->location); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
// op for some other operand. Update defining_ops. if (defining_ops.insert(new_control_input.getDefiningOp()).second) { operands.push_back(new_control_input); } } state.addOperands(operands); Operation* new_op = builder.create(state); item.replaceAllUsesWith(new_op); new_op->setAttrs(item.getAttrDictionary()); item.erase(); } AddStatefulControlFlowDependencies(graph_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0)