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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
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.dims(), 1)
    [all...]
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 77 milliseconds