diff options
author | 2017-06-19 10:48:05 -0700 | |
---|---|---|
committer | 2017-06-19 10:54:20 -0700 | |
commit | 46737e4e81314f7482bfd6a710f126a27f5d7975 (patch) | |
tree | 5e441f99cfe123f3b7a70ffcb79f7b308a5e04c8 /tensorflow/compiler/tf2xla/kernels/gather_op.cc | |
parent | 09d090282667e28ed04547ee0dcf1b6faba75659 (diff) |
Remove class xla::LiteralUtil. NFC (mind-numbingly so).
This patch removes class xla::LiteralUtil and rewrites every call to use class
xla::Literal instead.
PiperOrigin-RevId: 159446373
Diffstat (limited to 'tensorflow/compiler/tf2xla/kernels/gather_op.cc')
-rw-r--r-- | tensorflow/compiler/tf2xla/kernels/gather_op.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/compiler/tf2xla/kernels/gather_op.cc b/tensorflow/compiler/tf2xla/kernels/gather_op.cc index 49eadaf9d1..3c1cdef5f8 100644 --- a/tensorflow/compiler/tf2xla/kernels/gather_op.cc +++ b/tensorflow/compiler/tf2xla/kernels/gather_op.cc @@ -66,10 +66,10 @@ class GatherOp : public XlaOpKernel { std::vector<xla::ComputationDataHandle> args; args.push_back(tc.GetOrCreateRuntimeContextParameter()); args.push_back(b.ConstantLiteral( - *xla::LiteralUtil::CreateR0<int64>(indices_shape.num_elements()))); + *xla::Literal::CreateR0<int64>(indices_shape.num_elements()))); args.push_back(b.ConstantLiteral( - *xla::LiteralUtil::CreateR0<int64>(params_shape.dim_size(0)))); - args.push_back(b.ConstantLiteral(*xla::LiteralUtil::CreateR0<int64>( + *xla::Literal::CreateR0<int64>(params_shape.dim_size(0)))); + args.push_back(b.ConstantLiteral(*xla::Literal::CreateR0<int64>( params_shape.num_elements() / params_shape.dim_size(0)))); args.push_back(ctx->Input(0)); args.push_back(ctx->Input(1)); |