Searched refs:in0_shape (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | cross_op.cc | 28 TensorShape in0_shape = ctx->InputShape(0); variable 30 OP_REQUIRES(ctx, in0_shape == in1_shape, 32 in0_shape.DebugString(), " vs. ", 34 OP_REQUIRES(ctx, in0_shape.dims() >= 1, 36 in0_shape.DebugString())); 38 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); 49 std::vector<int64> starts(in0_shape.dims(), 0); 51 for (auto dim_size : in0_shape.dim_sizes()) { 54 std::vector<int64> strides(in0_shape [all...] |
H A D | mirror_pad_op.cc | 81 xla::StatusOr<std::unique_ptr<xla::Shape>> in0_shape = b->GetShape(in0); variable 82 OP_REQUIRES(ctx, in0_shape.ok(), in0_shape.status()); 84 DoMirrorPad(in0, *in0_shape.ValueOrDie(), pad_literal, b);
|
Completed in 359 milliseconds