Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 15 of 15 for _composite_op (0.16 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to top