Home
last modified time | relevance | path

Searched refs:xla_gather (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dgather_op_test.py337 def xla_gather(x, indices, axis): function
344 ("xla_gather", xla_gather)]
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py573 return gen_xla_ops.xla_gather(
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-no-tf2xla-fallback.mlir5966 // CHECK-LABEL: @xla_gather
5967 func.func @xla_gather(%arg0: tensor<200x100x300xf32>, %arg1: tensor<10x2xi32>) -> tensor<1x300x10xf…
Dlegalize-tf.mlir6135 // CHECK-LABEL: @xla_gather
6136 func.func @xla_gather(%arg0: tensor<200x100x300xf32>, %arg1: tensor<10x2xi32>) -> tensor<1x300x10xf…