Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_space_to_depth_pass.mlir116 …is_same_data_across_replicas, mhlo.sharding = "\08\01\1A\01\01\22\01\00"}, %arg3: tensor<64x1001xf…
117 …is_same_data_across_replicas, mhlo.sharding = "\08\01\1A\01\01\22\01\00"}, %arg3: tensor<64x1001xf…
190 …is_same_data_across_replicas, mhlo.sharding = "\08\01\1A\01\01\22\01\00"}, %arg3: tensor<64x1001xf…
191 …is_same_data_across_replicas, mhlo.sharding = "\08\01\1A\01\01\22\01\00"}, %arg3: tensor<64x1001xf…
Dannotate-parameter-replication.mlir22 // CHECK-SAME: %[[ARG1:.*]]: tensor<?xi32> {mhlo.is_same_data_across_replicas}
57 // CHECK-SAME: %[[ARG0:.*]]: tensor<?xi32> {mhlo.is_same_data_across_replicas},
59 …/ CHECK-SAME: %[[ARG2:.*]]: tensor<!tf.resource<tensor<?xi32>>> {mhlo.is_same_data_across_replicas}
81 // CHECK-NOT: mhlo.is_same_data_across_replicas
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_argument.cc49 if (is_same_data_across_replicas != other.is_same_data_across_replicas) { in operator ==()
Dxla_argument.h111 bool is_same_data_across_replicas = false; member
Dxla_compiler.cc441 &common, " is_same_data_across_replicas=", is_same_data_across_replicas); in HumanString()
1070 args[input_to_args->at(i)].is_same_data_across_replicas); in BuildArguments()
1103 args[input_to_args->at(i)].is_same_data_across_replicas); in BuildArguments()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport_replicas.mlir6 func @main(%arg0: tensor<16x16xf32>, %arg1: tensor<16x16xf32> {mhlo.is_same_data_across_replicas}) …
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/
Dconstant-folding.mlir4 …func @main(%arg0: tensor<10x19xf32>, %arg1: tensor<19x10xf32> {mhlo.is_same_data_across_replicas})…
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_op_util.cc45 if (arg.is_same_data_across_replicas()) { in CreateConfigPrefix()
Dtpu_compile_op_common.cc232 arg.is_same_data_across_replicas = proto_arg.is_same_data_across_replicas(); in BuildComputationArgumentDescriptions()
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dcompile_metadata.proto36 bool is_same_data_across_replicas = 5; field