- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for output_types (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
if (constant_y && constant_y.isSplat()) { APFloat y_value = constant_y.getSplatValue<APFloat>(); auto output_type = getType().cast<ShapedType>(); if (y_value.isZero() && output_type.hasStaticShape()) { return DenseElementsAttr::get( output_type, FloatAttr::get(output_type.getElementType(), /*value=*/1.0)); } if (y_value.isExactlyValue(1.0) && getX().getType() == getType()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
} for (auto edge : n->in_edges()) { inputs[index] = NodeDefBuilder::NodeOut{edge->src()->name(), edge->src_output(), edge->src()->output_type(edge->src_output())}; } } send_from_host_builder.Input(inputs); send_from_host_builder.Input(key_placeholder->name(), 0, DT_STRING); TF_RETURN_IF_ERROR(send_from_host_builder.Finalize(&send_from_host_def));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
// CHECK: "tfl.arg_max"(%arg0, %arg1) {output_type = 2 : i32} : (tensor<3xi32>, tensor<i32>) -> tensor<i32> %0 = "tfl.arg_max"(%arg0, %arg1) {output_type = 2 : i32} : (tensor<3xi32>, tensor<i32>) -> tensor<i32> func.return %0 : tensor<i32> } // ----- func.func @testArgMin(%arg0: tensor<3xi32>, %arg1: tensor<i32>) -> tensor<i32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
TFL_I32OrI64Tensor:$dim ); let results = (outs TFL_I32OrI64Tensor:$output ); let hasOptions = 1; DerivedTFLiteTypeAttr output_type = DerivedTFLiteTypeAttr<[{ return getResult().getType().cast<TensorType>().getElementType(). cast<IntegerType>().getWidth() > 32 ? tflite::TensorType_INT64 : tflite::TensorType_INT32; }], [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/c/c_api.cc
int TF_OperationNumOutputs(TF_Operation* oper) { return oper->node.num_outputs(); } TF_DataType TF_OperationOutputType(TF_Output oper_out) { return static_cast<TF_DataType>( oper_out.oper->node.output_type(oper_out.index)); } int TF_OperationOutputListLength(TF_Operation* oper, const char* arg_name, TF_Status* status) { NameRangeMap name_ranges; status->status =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0)