Home
last modified time | relevance | path

Searched refs:operand_segment_sizes (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dtf_device_ops_invalid.mlir64 }) {n = 2 : i32, operand_segment_sizes = array<i32: 0, 0>} : () -> ()
75 }) {n = 2 : i32, operand_segment_sizes = array<i32: 0, 0>} : () -> ()
87 }) {n = 2 : i32, operand_segment_sizes = array<i32: 0, 0>} : () -> ()
99 }) {n = 1 : i32, operand_segment_sizes = array<i32: 0, 0>} : () -> ()
111 …TPU_REPLICATED_CORE_0 = ["/DEVICE:0", "/DEVICE:1"]}, n = 3 : i32, operand_segment_sizes = array<i3…
123 …/DEVICE:1"], TPU_REPLICATED_CORE_1 = ["/DEVICE:2"]}, n = 3 : i32, operand_segment_sizes = array<i3…
134 …}) {n = 3 : i32, operand_segment_sizes = array<i32: 4, 0>} : (tensor<*xi32>, tensor<*xi32>, tensor…
146 …}) {n = 3 : i32, operand_segment_sizes = array<i32: 3, 2>} : (tensor<*xi32>, tensor<*xi32>, tensor…
158 }) {n = 2 : i32, operand_segment_sizes = array<i32: 2, 0>} : (tensor<*xi32>, tensor<*xi1>) -> ()
170 }) {n = 2 : i32, operand_segment_sizes = array<i32: 0, 1>} : (tensor<*xi1>) -> ()
[all …]
H A Drewrite_tpu_embedding_ops.mlir18 …XlaSendTPUEmbeddingGradients"(%arg0, %[[DATA]]) {config = {{.*}}, operand_segment_sizes = array<i3…
22 … \01*\0C\1A\00j\05\0D\00\00\80?\88\01\01\10\02\18\80\04 \01(\02", operand_segment_sizes = array<i3…
33 … \01*\0C\1A\00j\05\0D\00\00\80?\88\01\01\10\02\18\80\04 \01(\02", operand_segment_sizes = array<i3…
H A Dtf_device_ops.mlir100 tf_device.replicate {n = 2 : i32, operand_segment_sizes = array<i32: 0, 0>} {
106 // CHECK-NOT: operand_segment_sizes
H A Dtpu_update_embedding_enqueue_op_inputs.mlir44 …%2, %3) {_tpu_embedding_layer = "call1", config = "\0A\0B\0C\0D", operand_segment_sizes = array<i3…
H A Dside-effect-analysis-test.mlir1264 …t_func = @__func_c, next_func.experimental_ints_on_device = true, operand_segment_sizes = array<i3…
1270 …t_func = @__func_c, next_func.experimental_ints_on_device = true, operand_segment_sizes = array<i3…
1339 …t_func = @__func_c, next_func.experimental_ints_on_device = true, operand_segment_sizes = array<i3…
1618 …t_func = @__func_c, next_func.experimental_ints_on_device = true, operand_segment_sizes = array<i3…
1621 …t_func = @__func_c, next_func.experimental_ints_on_device = true, operand_segment_sizes = array<i3…
1652 …t_func = @__func_c, next_func.experimental_ints_on_device = true, operand_segment_sizes = array<i3…
2151 …t_func = @__func_c, next_func.experimental_ints_on_device = true, operand_segment_sizes = array<i3…
2303 …t_func = @__func_c, next_func.experimental_ints_on_device = true, operand_segment_sizes = array<i3…
H A Dbreakup-islands.mlir470 operand_segment_sizes = array<i32: 2, 2, 1>,
482 operand_segment_sizes = array<i32: 2, 2, 1>,
H A Dtpu_extract_head_tail_outside_compilation.mlir573 …N = 1 : i64, NN = 0 : i64, config = "test_config_send_embedding", operand_segment_sizes = array<i3…
598 …N = 1 : i64, NN = 0 : i64, config = "test_config_send_embedding", operand_segment_sizes = array<i3…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/gml_st/
H A Dinvalid.mlir78 operand_segment_sizes = array<i32: 2, 2, 2, 2, 2>
100 operand_segment_sizes = array<i32: 1, 1, 1, 0, 1>
170 …offsets = [0, 0], static_sizes = [1, 1], static_strides = [1, 1], operand_segment_sizes = array<i3…
179 …offsets = [0, 0], static_sizes = [8, 8], static_strides = [1, 1], operand_segment_sizes = array<i3…
188 …offsets = [0, 0], static_sizes = [8, 8], static_strides = [1, 1], operand_segment_sizes = array<i3…
197 …fsets = [0, 0], static_sizes = [-1, -2], static_strides = [1, 1], operand_segment_sizes = array<i3…
206 …fsets = [0, 0], static_sizes = [-1, 8], static_strides = [1, -2], operand_segment_sizes = array<i3…
215 …fsets = [0, -2], static_sizes = [-1, 8], static_strides = [1, 1], operand_segment_sizes = array<i3…
224 …sets = [0, 32], static_sizes = [-1, -1], static_strides = [1, 1], operand_segment_sizes = array<i3…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_device.cc389 operand_segment_sizes()[0]; in print()
466 auto operand_segment_sizes = in BuildReplicateOp() local
468 state->addAttribute(kOperandSegmentSizesAttr, operand_segment_sizes); in BuildReplicateOp()
504 auto operand_segment_sizes = op.operand_segment_sizes(); in verify() local
506 operand_segment_sizes[0]; in verify()
508 operand_segment_sizes[1]; in verify()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/
H A Dfuse_tpu_compile_and_execute_ops.mlir14 …"(%arg0, %0) {metadata = "metadata", mlir_module = "mlir_module", operand_segment_sizes = array<i3…
41 …"(%arg0, %0) {metadata = "metadata", mlir_module = "mlir_module", operand_segment_sizes = array<i3…
72 …arg4, %arg3) {metadata = "metadata", mlir_module = "mlir_module", operand_segment_sizes = array<i3…
73 … %2, %0, %1) {metadata = "metadata", mlir_module = "mlir_module", operand_segment_sizes = array<i3…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
H A Dbatch_function_deduplicate.mlir10 … max_enqueued_batches = 10000 : i64, num_batch_threads = 2 : i64, operand_segment_sizes = array<i3…
25 … max_enqueued_batches = 10000 : i64, num_batch_threads = 2 : i64, operand_segment_sizes = array<i3…
39 … max_enqueued_batches = 10000 : i64, num_batch_threads = 2 : i64, operand_segment_sizes = array<i3…
H A Dbatch_function_deduplicate_failed.mlir7 … max_enqueued_batches = 10000 : i64, num_batch_threads = 2 : i64, operand_segment_sizes = array<i3…
16 … max_enqueued_batches = 10000 : i64, num_batch_threads = 2 : i64, operand_segment_sizes = array<i3…
H A Dbatch_function_fallback_resource_variable_as_captured_tensor.mlir14 …ed_func, max_batch_size = 256 : i64, num_batch_threads = 2 : i64, operand_segment_sizes = array<i3…
H A Dbatch_function_fallback.mlir28 …64, max_enqueued_batches = 10 : i64, num_batch_threads = 1 : i64, operand_segment_sizes = array<i3…
H A Dlower_saved_model.mlir149 …64, max_enqueued_batches = 10 : i64, num_batch_threads = 1 : i64, operand_segment_sizes = dense<1>…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/lhlo_to_jitrt/
H A Dcustom_call_to_jitrt.mlir30 operand_segment_sizes = array<i32: 0, 1>
64 operand_segment_sizes = array<i32: 2, 3>,
H A Dpad_to_static_to_gpu_binary.mlir40 operand_segment_sizes = array<i32: 1, 2>
H A Dslice_to_dynamic_to_gpu_binary.mlir40 operand_segment_sizes = array<i32: 2, 1>
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/lhlo/
H A Dops.mlir1088 operand_segment_sizes = array<i32: 2, 2>,
1107 operand_segment_sizes = array<i32: 2, 2>,
1126 operand_segment_sizes = array<i32: 2, 2>,
1145 operand_segment_sizes = array<i32: 2, 2>,
1164 operand_segment_sizes = array<i32: 2, 2>,
1183 operand_segment_sizes = array<i32: 2, 2>,
1202 operand_segment_sizes = array<i32: 2, 2>,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
H A Dparse_example.pbtxt4 # CHECK: operand_segment_sizes = array<i32: 1, 1, 0, 1, 1>
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
H A Dtf_tpu_embedding_ops.mlir6 …N = 1 : i64, NN = 0 : i64, config = "test_config_send_embedding", operand_segment_sizes = array<i3…
H A Dparse_example.mlir56 …_0, %outputs_2) {dense_shapes = [#tf_type.shape<1>], device = "", operand_segment_sizes = array<i3…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
H A Dlift_tflite_flex_ops.mlir57 // CHECK-SAME: operand_segment_sizes = array<i32: 1, 1, 0, 2, 2>, result_segment_sizes = array<i32:…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
H A Dhlo-legalize-to-memref.mlir124 …d_config = "", call_target_name = "foo", has_side_effect = false, operand_segment_sizes = array<i3…
H A Dhlo-legalize-to-lhlo.mlir622 …d_config = "", call_target_name = "foo", has_side_effect = false, operand_segment_sizes = array<i3…
634 …d_config = "", call_target_name = "foo", has_side_effect = false, operand_segment_sizes = array<i3…

12