Searched refs:XlaGather (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 35 Status XlaGather(const xla::XlaOp& input, const TensorShape& input_shape, in XlaGather() function
228 XlaGather(input, input_shape, indices, indices_shape, *axis, in XlaGatherWithBatchDimsOpImpl()
294 OP_REQUIRES_OK(context, XlaGather(params, params_shape, indices, in Compile()
|
| D | gather_op_helpers.h | 36 Status XlaGather(const xla::XlaOp& input, const TensorShape& input_shape,
|
| D | random_ops.cc | 206 OP_REQUIRES_OK(ctx, XlaGather(input, input_shape, swapped_indices, in Compile()
|
| D | tensor_list_ops.cc | 416 ctx, XlaGather(buffer, buffer_shape, indices, indices_shape, /*axis=*/0, in Compile()
|
| D | tensor_array_ops.cc | 342 XlaGather(ta, ta_shape, indices, indices_shape, /*axis=*/0, in Compile()
|
| D | image_ops.cc | 569 XlaGather(indices_sorted, scores_shape, selected_indices_sorted, in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 740 // XlaGather op.
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-tf.mlir | 713 …%2 = "tf.XlaGather"(%arg0, %0, %1) {device = "", dimension_numbers = "\0A\04\00\01\02\03\1A\01\02"…
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 5114 …%0 = "tf.XlaGather"(%arg0, %arg1, %cst) {dimension_numbers = "\0A\01\01\12\01\00\1A\01\00 \01", in…
5131 …%0 = "tf.XlaGather"(%arg0, %arg1, %cst) {dimension_numbers = "\0A\01\01\12\01\00\1A\01\00 \01", in…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 29391 func XlaGather(scope *Scope, operand tf.Output, start_indices tf.Output, slice_sizes tf.Output, dim… func
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17793 def TF_XlaGatherOp : TF_Op<"XlaGather", [NoSideEffect]> {
|