Home
last modified time | relevance | path

Searched refs:collective_permute (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py113 def collective_permute(x, source_target_pairs, name=None): function
136 return gen_tpu_ops.collective_permute(x, source_target_pairs, name=name)
145 return [gen_tpu_ops.collective_permute(grad, source_target_pairs), None]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_compile_test.cc30 …"lmhlo.collective_permute"(%arg0, %arg1) {source_target_pairs = dense<[[0, 1, 2]]> : tensor<1x3xi6… in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dops.mlir317 %0 = "mhlo.collective_permute"(%arg0) {
327 %0 = "mhlo.collective_permute"(%arg0) {
337 %0 = "mhlo.collective_permute"(%arg0) {
347 %0 = "mhlo.collective_permute"(%arg0) {
Dlhlo_ops.mlir778 "lmhlo.collective_permute"(%arg0, %arg_out) {
782 "lmhlo.collective_permute"(%arg0, %arg_out) {
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td579 def LHLO_CollectivePermuteOp: LHLO_Op<"collective_permute", [SameTypeOperands]>,
Dhlo_ops.td893 def HLO_CollectivePermuteOp: HLO_Op<"collective_permute",
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt633 // CHECK: "lmhlo.collective_permute"
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir256 %0 = "mhlo.collective_permute"(%arg0) {
Dimport.hlotxt137 …// CHECK-NEXT: "mhlo.collective_permute"([[ARG]]) {source_target_pairs = dense<{{\[\[}}0, 1], [1,…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir3850 // CHECK-LABEL: @collective_permute
3851 func @collective_permute(%arg0: tensor<128x32xf32>) -> tensor<128x32xf32> {
3856 // CHECK: "mhlo.collective_permute"