- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for one_output (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir
// CHECK: %[[D_OUTPUT:[0-9]*]] = "tf.D"(%[[C_OUTPUT]]) // CHECK: %[[E_OUTPUT:[0-9]*]]:2 = "tf.E"(%[[D_OUTPUT]]) // CHECK: tf_device.return %[[D_OUTPUT:[0-9]*]], %[[E_OUTPUT:[0-9]*]]#0, %[[E_OUTPUT:[0-9]*]]#1 // CHECK: "tf_device.cluster" // CHECK: %[[A_OUTPUT:[0-9]*]] = "tf.A"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 129.6K bytes - Viewed (0) -
tensorflow/c/c_api.cc
std::vector<tensorflow::Output> cond_output; TF_RETURN_IF_ERROR(CopyGraph( ¶ms->cond_graph->graph, &parent->graph, &parent->refiner, params->cond_inputs, inputs, scope.impl()->name(), scope.impl()->control_deps(), ¶ms->cond_output, /* nreturn_nodes */ 1, &cond_output)); *output = cond_output[0]; return absl::OkStatus(); };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
int64_t time = time_majored ? input_type.getDimSize(0) : input_type.getDimSize(1); int64_t n_output = output_state_type.getDimSize(1); // Build the output shape. SmallVector<int64_t, 3> output_shape; if (time_majored) { output_shape = {time, batch, n_output}; } else { output_shape = {batch, time, n_output}; } auto result_type = tensorflow::GetTypeFromTFTensorShape(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
// Fuse Mul with proceeding Affine ops. This is an C++ implementation of the // following table gen implementation, which doesn't derived the result type of // the TFL_DequantizeOp. // def : Pat<(TFL_MulOp (TFL_Conv2DOp:$conv_output $input, // (TFL_DequantizeOp (TFL_QuantizeOp // (Arith_ConstantOp F32ElementsAttr:$filter), // $qtype)),
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/tensorflow/translate/import_model.cc
// result of the new operation, and deletes the old operation. for (unsigned i = 0, e = dst->getNumResults(); i != e; ++i) { auto new_output = new_dst->getResult(i); dst->getResult(i).replaceAllUsesWith(new_output); } dst->dropAllReferences(); dst->erase(); return absl::OkStatus(); } absl::StatusOr<mlir::FunctionType> ImporterBase::InferLibFunctionType(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)