Home
last modified time | relevance | path

Searched defs:gather_indices (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc138 XlaOp Gather2by2Neighbors(xla::XlaBuilder* b, XlaOp data, XlaOp gather_indices, in Gather2by2Neighbors()
249 XlaOp gather_indices, XlaOp warp, in CalculateGradData()
360 XlaOp gather_indices, XlaOp data, in CalculateGradWarp()
519 auto gather_indices = ConcatenateIota(ctx->builder(), top_left, warp_shape); in Compile() local
656 auto gather_indices = ConcatenateIota(ctx->builder(), top_left, warp_shape); in Compile() local
/external/tensorflow/tensorflow/compiler/tests/
Dgather_test.py161 def _benchmarkGather(self, name, axis, gather_indices, use_xla_jit): argument
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing.cc214 XlaOp gather_indices = ConcatInDim(builder, to_concat, input_shape.rank()); in TorchGather() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc655 HloInstruction* gather_indices = comp->AddInstruction( in RewriteDynamicReshapeCombineInput() local