Searched refs:inp0_rank (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | concat_op.cc | 159 const int64 inp0_rank = inp0_shape.num_elements(); in Compile() local
164 VLOG(1) << "ConcatOffset " << cdim << "," << inp0_rank; in Compile()
165 int32 axis = cdim < 0 ? cdim + inp0_rank : cdim; in Compile()
166 OP_REQUIRES(ctx, FastBoundsCheck(axis, inp0_rank), in Compile()
168 " vs. ", inp0_rank)); in Compile()
172 OP_REQUIRES(ctx, inp0_rank == inp_shape.num_elements(), in Compile()
174 inp0_rank, " elements, but got ", in Compile()
179 Tensor out_constant(DT_INT32, TensorShape({inp0_rank})); in Compile()
181 for (int64 j = 0; j < inp0_rank; ++j) { in Compile()
|