- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for num_inputs (0.23 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
FunctionType func_type = func.getFunctionType(); int num_inputs = func_type.getNumInputs(); int num_results = func_type.getNumResults(); // For each argument type in function's arguments, change it to uranked // tensor type if it's a variant type. SmallVector<Type, 8> updated_argument_types; updated_argument_types.reserve(num_inputs); UpdateTensorListTypes<mlir::OperandRange>(
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/tf2xla/transforms/legalize_tf.cc
? mhlo::Precision::DEFAULT : mlir::mhlo::Precision::HIGHEST; llvm::SmallVector<mlir::Attribute, 2> attr_vec; const int num_inputs = 2; for (int i = 0; i < num_inputs; i++) { attr_vec.push_back( mlir::mhlo::PrecisionAttr::get(builder->getContext(), precision)); } return builder->getArrayAttr(attr_vec); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/views/op_view.cc
const std::vector<AttrView>& OpView::Attributes() const { return argument_attrs_; } const std::vector<OpArgumentView>& OpView::AllArguments() const { return all_arguments_; } int OpView::NumInputs() const { return input_args_.size(); } int OpView::NumOutputs() const { return output_args_.size(); } ArgView OpView::OnlyInput() const { CHECK_EQ(input_args_.size(), 1); // Crash OK return input_args_.front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 07:02:00 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td
def LowerSoftmaxCrossEntropyWithLogitsOp : Pattern< (TF_SoftmaxCrossEntropyWithLogitsOp AnyRankedTensor:$features, AnyRankedTensor:$labels), [(TF_SumOp (TF_MulNoNanOp:$sum_input (TF_LogSoftmaxOp $features), (TF_NegOp $labels)), (TF_ConstOp (GetI64ScalarElementsAttr<-1>)), /*keep_dims=*/ConstBoolAttrFalse),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0)