- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for _composite_op (0.16 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_custom_call_to_composite.cc
return op->emitError("expected exactly one called_computation"); auto decomposition = mlir::cast<FlatSymbolRefAttr>(calledComputations[0]); auto composite = rewriter.create<mlir::stablehlo::CompositeOp>( op.getLoc(), op.getResultTypes(), op.getOperands(), name.str(), attrs, decomposition.getValue()); rewriter.replaceOp(op, composite.getResults()); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/README.md
## RFC This project is an alternative implementation of [RFC:Standardizing composite ops in tensorflow to support efficient inference](https://github.com/tensorflow/community/blob/master/rfcs/20190610-standardizing-composite_ops.md). This project doesn't rely on the tracing functionality provided by `tf.function` to avoid all its pitfalls and it helps to build more general transformations in the backends. ## Team * Feng Liu
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 29 18:32:13 UTC 2022 - 6.2K bytes - Viewed (0) -
tensorflow/cc/framework/cc_ops_test.cc
auto c = Concat(root, {p[0], p[1]}, 0); TF_EXPECT_OK(root.status()); Tensor out; test::GetTensor(root, c, &out); test::ExpectTensorEqual<int>(out, test::AsTensor<int>({1, 2}, {2, 1})); } TEST(CCOpTest, CompositeOp) { Scope root = Scope::NewRootScope(); auto l = Linear(root.WithOpName("layer0"), {{10.0f, -3.0f}}, {{.8f, .5f}, {.1f, .6f}}, {-8.0f, 31.0f}); TF_EXPECT_OK(root.status());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 15 15:13:38 UTC 2023 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
Translator::BuildVhloCompositeV1Op(mlir::vhlo::CompositeOpV1 composite_op, const std::vector<int32_t>& operands, const std::vector<int32_t>& results, std::string op_name) { uint32_t opcode_index = GetOpcodeIndex(op_name, tflite::BuiltinOperator_STABLEHLO_COMPOSITE); int32_t api_version = composite_op.getVersion()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
// CHECK-NEXT: %0:2 = "tfl.custom"(%arg0, %arg1, %arg2) <{custom_code = "my_composite_op", custom_option = #tfl<const_bytes : "0x6578616D706C655F6F7074696F6E006578616D706C655F737472000A76616C756520312E30310002281A0201020A120414042401">}> : (tensor<4x4xf32>,...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0)