- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 1,881 for Levine (0.15 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/tests/e2e/device-transform-nnapi.mlir
// RUN: tac-translate -input-mlir -output-mlir -device-specs=NNAPI %s -o - 2>&1 | FileCheck %s module { // CHECK-LABEL: main func.func @main(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>) -> tensor<4xf32> { %0 = "tfl.squared_difference"(%arg0, %arg1) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32> func.return %0 : tensor<4xf32> // CHECK: [[VAL_0:%.*]] = tfl.sub %arg0, %arg1 {fused_activation_function = "NONE"} : tensor<4xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/device-arg-retval-attr.mlir
// Verify arg/ret attributes are exported as device assignment for arg/retval // nodes. module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 121 : i32}} { func.func @main(%arg0: tensor<*xf32> {tf.device = "/CPU:0"}, %arg1: tensor<2x4x6x8xi32>) -> (tensor<*xf32>, tensor<2x4x6x8xi32> {tf.device = "/CPU:1"})
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/device-transform-nnapi.mlir
// RUN: tac-opt-all-backends -tfl-device-transform-nnapi %s -split-input-file -verify-diagnostics | FileCheck %s func.func @mean_4d_keepdim(%arg0: tensor<1x48x48x512xf32>) -> tensor<1x1x1x512xf32> { %cst = arith.constant dense<[1, 2]> : tensor<2xi32> %0 = "tfl.mean"(%arg0, %cst) {keep_dims = true} : (tensor<1x48x48x512xf32>, tensor<2xi32>) -> tensor<1x1x1x512xf32> func.return %0 : tensor<1x1x1x512xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h
#define GEN_PASS_DECL_TPUCLUSTERFORMATIONPASS #define GEN_PASS_DECL_TPUEXTRACTHEADTAILOUTSIDECOMPILATIONPASS #define GEN_PASS_DECL_TPUEXTRACTOUTSIDECOMPILATIONPASS #define GEN_PASS_DECL_TPUSHARDINGIDENTIFICATIONPASS #define GEN_PASS_DECL_VERIFYCLUSTERINGPASS #define GEN_PASS_DECL_XLACLUSTERFORMATIONPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" } // namespace internal } // namespace tf2xla
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/passes.h
#define GEN_PASS_DECL_LEGALIZETF #define GEN_PASS_DECL_LEGALIZETFCOLLECTIVE #define GEN_PASS_DECL_LEGALIZETFMODULEPASS #define GEN_PASS_DECL_LEGALIZETFTYPESPASS #define GEN_PASS_DECL_TFXLADEVICESPECIFICTRANSFORMS #define GEN_PASS_DECL_VERIFYTFXLALEGALIZATION #include "tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_passes.h.inc" #define GEN_PASS_REGISTRATION #define GEN_PASS_DECL_LEGALIZETFCOMMUNICATIONPASS
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/aot/test_google.cc
#include "tensorflow/core/platform/cpu_info.h" // Macros that expand to tokens based on the entry point name. // clang-format off #define CPP_CLASS {{TFCOMPILE_CPP_CLASS}} // NOLINT(whitespace/braces) #define TEST_NAME {{TFCOMPILE_NAME}}Test // NOLINT(whitespace/braces) #define BM_NAME BM_{{TFCOMPILE_NAME}} // NOLINT(whitespace/braces) // clang-format on namespace tensorflow { namespace tfcompile { namespace {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/aot/test.cc
#include "tensorflow/core/platform/test_benchmark.h" // Macros that expand to tokens based on the entry point name. // clang-format off #define CPP_CLASS {{TFCOMPILE_CPP_CLASS}} // NOLINT(whitespace/braces) #define TEST_NAME {{TFCOMPILE_NAME}}Test // NOLINT(whitespace/braces) #define BM_NAME BM_{{TFCOMPILE_NAME}} // NOLINT(whitespace/braces) // clang-format on namespace tensorflow { namespace tfcompile { namespace {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_assignment.cc
} }); } }; #define GEN_PASS_DEF_TFDEVICEASSIGNMENTBYFUNCATTRPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" // A pass to perform device assignment for TF dialect ops that do not // have device assignment, by using the device attribute of the function. // If device attribute is not found from the function, nothing is done. class TFDeviceAssignmentByFuncAttrPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/aot/benchmark_main.template
#define EIGEN_USE_THREADS #define EIGEN_USE_CUSTOM_THREAD_POOL // clang-format off #include "{{TFCOMPILE_HEADER}}" // NOLINT(whitespace/braces) // clang-format on #include "tensorflow/compiler/aot/benchmark.h" #include "unsupported/Eigen/CXX11/Tensor" // Macros that expand to tokens based on the entry point name. // clang-format off
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 19 20:05:05 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.h
// `container`, `name`, and `device` can be set. If an resource handle is tied // to an instance of an operation (e.g. TensorFlow runtime operation caching), // `op` can be set instead. struct ResourceHandle { ResourceHandle(StringRef container, StringRef name, StringRef device, Operation* op) : container(container), name(name), device(device), op(op) {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 03 19:26:14 UTC 2023 - 6.5K bytes - Viewed (0)