Searched refs:then_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | select_op.cc | 35 const TensorShape then_shape = ctx->InputShape(1); in Compile() local
39 ctx, then_shape.IsSameSize(else_shape), in Compile()
42 then_shape.DebugString(), " vs. ", else_shape.DebugString())); in Compile()
48 bool broadcasting = !cond_shape.IsSameSize(then_shape); in Compile()
55 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then_shape), in Compile()
58 then_shape.DebugString())); in Compile()
59 OP_REQUIRES(ctx, then_shape.dim_size(0) == cond_shape.num_elements(), in Compile()
62 then_shape.dim_size(0), " vs. ", in Compile()
68 const auto dim_sizes = then_shape.dim_sizes(); in Compile()
73 std::vector<int64> dim_order(then_shape.dims()); in Compile()
[all …]
|