Home
last modified time | relevance | path

Searched refs:XlaGatherWithBatchDimsOpImpl (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op_helpers.h45 Status XlaGatherWithBatchDimsOpImpl(XlaOpKernelContext* context,
Dgather_op.cc155 Status XlaGatherWithBatchDimsOpImpl(XlaOpKernelContext* context, in XlaGatherWithBatchDimsOpImpl() function
251 XlaGatherWithBatchDimsOpImpl(context, input, input_shape, in Compile()
Dvariable_ops.cc137 OP_REQUIRES_OK(ctx, XlaGatherWithBatchDimsOpImpl(ctx, input, input_shape, in Compile()