Searched defs:gather_indices (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 141 XlaOp Gather2by2Neighbors(xla::XlaBuilder* b, XlaOp data, XlaOp gather_indices, in Gather2by2Neighbors()
252 XlaOp gather_indices, XlaOp warp, in CalculateGradData()
363 XlaOp gather_indices, XlaOp data, in CalculateGradWarp()
521 auto gather_indices = ConcatenateIota(ctx->builder(), top_left, warp_shape); in Compile() local
654 auto gather_indices = ConcatenateIota(ctx->builder(), top_left, warp_shape); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | gather_test.py | 157 def _benchmarkGather(self, name, axis, gather_indices, use_xla_jit): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 215 XlaOp gather_indices = ConcatInDim(builder, to_concat, input_shape.rank()); in TorchGather() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| D | layer_utils.h | 449 nvinfer1::ITensor* gather_indices = (*indices_result)->getOutput(0); variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 710 HloInstruction* gather_indices = reshape->AddInstruction( in RewriteDynamicReshapeCombineInput() local
|