HomeSort by relevance Sort by last modified time
    Searched refs:ConstantInputReshaped (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
fill_op.cc 47 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshaped(
pad_op.cc 58 ctx, ctx->ConstantInputReshaped(1, {fixed_dims, 2}, &pad_literal));
mirror_pad_op.cc 77 ctx, ctx->ConstantInputReshaped(1, {fixed_dims, 2}, &pad_literal));
reverse_op.cc 53 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshaped(1, {x_shape.dims()}, &lax));
transpose_op.cc 54 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshaped(1, {dims}, &literal));
reduction_ops_common.cc 65 ctx->ConstantInputReshaped(
shape_op.cc 122 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshaped(1, {1}, &literal));
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_op_kernel.h 104 Status ConstantInputReshaped(int index, gtl::ArraySlice<int64> new_shape,
xla_op_kernel.cc 74 return ConstantInputReshaped(
78 Status XlaOpKernelContext::ConstantInputReshaped(
98 return errors::Internal("Incompatible shapes in ConstantInputReshaped.");

Completed in 107 milliseconds