- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for TF_MLIR_TFR_LIB_DIR (0.23 sec)
-
tensorflow/compiler/mlir/tfr/examples/customization/test_ops_test.py
def test_test_ops(self): attr = tf.function(test_ops.test_attr)(tf.float32) self.assertAllClose(attr.numpy(), 100.0) if __name__ == '__main__': os.environ['TF_MLIR_TFR_LIB_DIR'] = ( 'tensorflow/compiler/mlir/tfr/examples/customization')
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 17 22:41:04 UTC 2023 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/node_expansion_test.py
self.assertAllEqual( x, gen_resource_variable_ops.ReadVariableOp( resource=handle, dtype=dtypes.float32)) if __name__ == '__main__': os.environ['TF_MLIR_TFR_LIB_DIR'] = 'tensorflow/compiler/mlir/tfr/resources' ops.enable_eager_execution()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/pad/pad_ops_test.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/graph_decompose_test.py
t3 = constant_op.constant([[-10.0, -10.0], [-10.0, -10.0]]) sq = biasd_dense_elu(t1, t2, t3) self.assertAllClose(sq.numpy().reshape(-1), [-0.950213, 0, 5, 12]) if __name__ == '__main__': os.environ['TF_MLIR_TFR_LIB_DIR'] = 'tensorflow/compiler/mlir/tfr/resources' ops.enable_eager_execution()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/mnist_ops_test.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/mnist_train.py
def main(strategy): """Trains an MNIST model using the given tf.distribute.Strategy.""" # TODO(fengliuai): put this in some automatically generated code. os.environ[ 'TF_MLIR_TFR_LIB_DIR'] = 'tensorflow/compiler/mlir/tfr/examples/mnist' ds_train = tfds.load('mnist', split='train', shuffle_files=True) ds_train = ds_train.shuffle(1024).batch(batch_size).prefetch(64)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 20 03:05:18 UTC 2021 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
#include "tensorflow/core/platform/stringpiece.h" #include "tensorflow/core/util/env_var.h" #include "tsl/platform/statusor.h" namespace tensorflow { namespace tfr { const char* const kTFRLibEnv = "TF_MLIR_TFR_LIB_DIR"; absl::StatusOr<std::unique_ptr<TFRDecomposeContext>> TFRDecomposeContext::Get( mlir::MLIRContext* mlir_ctx) { Env* env = Env::Default(); std::string tfr_lib_dir; TF_RETURN_IF_ERROR(ReadStringFromEnvVar(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0)