- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for AddN (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
// CHECK-NEXT: %[[ADDN:.*]] = "tf.AddN"(%[[UPDATE]], %[[BROADCAST]]) : (tensor<10xf32>, tensor<10xf32>) -> tensor<10xf32> %addn = "tf.AddN"(%set, %tl) : (tensor<!tf_type.variant<tensor<f32>>>, tensor<!tf_type.variant<tensor<f32>>>) -> tensor<!tf_type.variant<tensor<f32>>> // CHECK-NEXT: %[[ZEROS_LIKE:.*]] = "tf.ZerosLike"(%[[ADDN]]) : (tensor<10xf32>) -> tensor<10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
return failure(); } } else if (auto addn = llvm::dyn_cast<TF::AddNOp>(&op)) { auto it = buffer_to_size->find(addn.getOperand(0)); if (it != buffer_to_size->end()) { addn.getSum().setType( mlir::cast<TensorType>(addn.getOperand(0).getType())); auto size = it->getSecond(); (*buffer_to_size)[addn.getSum()] = size; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler_test.cc
TF_InferStatically(graph_properties, true, false, false, false, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); for (const NodeDef& node : item.graph.node()) { if (node.op() == "AddN") { int num_values = 0; TF_GetInputPropertiesListSize(graph_properties, node.name().c_str(), &num_values, status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 22:30:58 UTC 2023 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
}); return has_variant_input || has_variant_output; } // There are 2 standard tf ops which are not TensorList ops that may take as // input a tensorlist. These are tf.AddN and tf.ZeroesLike. Since the runtime // implementation of a tensorlist are not compatible between tf and tflite // we cannot use tflite tensorlist kernels until these cases are handled.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference_test.cc
auto c = ops::Placeholder(root.WithOpName("C"), DT_FLOAT); auto d = ops::Add(root.WithOpName("D"), a, b); auto e = ops::Add(root.WithOpName("E"), d, c); auto f = ops::Neg(root.WithOpName("F"), e); auto g = ops::AddN(root.WithOpName("G"), std::initializer_list<Output>{e, f}); std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); TF_CHECK_OK(root.ToGraph(graph.get())); GraphShapeInfo shape_info;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
// CHECK: [[d:%.*]] = tf_mlrt.await [[futures]]#2 %d = "tf_mlrt.tf_await"(%future_d) : (!mlrt.future) ->tensor<i32> // CHECK: [[result:%.*]] = tf_mlrt.executeop([[b]], [[d]], [[f]]) // CHECK-SAME: AddN %result = "tf.AddN"(%b, %d, %f) {__op_key = 9: i32}: (tensor<i32>, tensor<i32>, tensor<i32>) -> tensor<i32> // CHECK: mlrt.await_handle [[handle_0]] // CHECK: mlrt.await_handle [[handle_1]] mlrt.await_handle %handle_0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/cc/gradients/math_grad_test.cc
RunTest({x}, {x_shape}, {y}, {y_shape}); } TEST_F(NaryGradTest, AddN) { TensorShape shape({3, 2, 5}); std::vector<Output> xs; xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape))); xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape))); xs.push_back(Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape))); auto y = AddN(scope_, xs); RunTest(xs, {shape, shape, shape}, {y}, {shape}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 36K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
if (gradient_tensors.size() == 1) { return gradient_tensors[0]; } AbstractOperationPtr op(ctx_->CreateOperation()); Status s = op->Reset("AddN", /*raw_device_name=*/nullptr); if (!s.ok()) { return nullptr; } s = op->AddInputList(gradient_tensors); if (!s.ok()) { return nullptr; } int num_outputs = 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir
} // CHECK-LABEL: unsupported_dtype func.func @unsupported_dtype(%arg0: tensor<2x!tf_type.variant>) -> tensor<2x!tf_type.variant> { // CHECK: tf.AddN // expected-remark@+1 {{skipping legalization due to unsupported type 'tensor<2x!tf_type.variant>'}} %0 = "tf.AddN"(%arg0, %arg0) : (tensor<2x!tf_type.variant>, tensor<2x!tf_type.variant>) -> tensor<2x!tf_type.variant> func.return %0 : tensor<2x!tf_type.variant> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/parallelization.mlir
%b2 = "tf.Sub"(%b1, %b) : (tensor<i32>, tensor<i32>) -> tensor<i32> %b3 = "tf.Sub"(%b2, %b) : (tensor<i32>, tensor<i32>) -> tensor<i32> %d = "tf.AddN"(%a3, %b3, %c) : (tensor<i32>, tensor<i32>, tensor<i32>) -> tensor<i32> func.return %d : tensor<i32> } // ----- // Test that constants are copied instead of using promise/await.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 22:07:30 UTC 2023 - 15K bytes - Viewed (0)