Searched refs:in0_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cross_op.cc | 29 TensorShape in0_shape = ctx->InputShape(0); in Compile() local
31 OP_REQUIRES(ctx, in0_shape == in1_shape, in Compile()
33 in0_shape.DebugString(), " vs. ", in Compile()
35 OP_REQUIRES(ctx, in0_shape.dims() >= 1, in Compile()
37 in0_shape.DebugString())); in Compile()
39 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); in Compile()
50 std::vector<int64> starts(in0_shape.dims(), 0); in Compile()
52 for (auto dim_size : in0_shape.dim_sizes()) { in Compile()
55 std::vector<int64> strides(in0_shape.dims(), 1); in Compile()
76 auto output = xla::ConcatInDim(b, {s1, s2, s3}, in0_shape.dims() - 1); in Compile()
|
| D | mirror_pad_op.cc | 94 xla::StatusOr<xla::Shape> in0_shape = b->GetShape(in0); in Compile() local
95 OP_REQUIRES(ctx, in0_shape.ok(), in0_shape.status()); in Compile()
97 DoMirrorPad(in0, in0_shape.ValueOrDie(), pad_literal, mode, b); in Compile()
|