Searched refs:pos_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | substr_op.cc | 53 const TensorShape& pos_shape = pos_tensor.shape(); in Compute() local
55 OP_REQUIRES(context, (pos_shape == len_shape), in Compute()
58 pos_shape.DebugString(), " vs. ", len_shape.DebugString())); in Compute()
60 bool is_scalar = TensorShapeUtils::IsScalar(pos_shape); in Compute()
62 if (is_scalar || input_shape == pos_shape) { in Compute()
136 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(pos_shape)); in Compute()
140 " vs. ", pos_shape.DebugString())); in Compute()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | string_ops.cc | 261 ShapeHandle pos_shape = c->input(1); in __anonf38d6eaa0702() local
266 TF_RETURN_IF_ERROR(c->WithRank(pos_shape, c->Rank(len_shape), &unused)); in __anonf38d6eaa0702()
269 for (int32 i = 0; i < c->Rank(pos_shape); ++i) { in __anonf38d6eaa0702()
270 DimensionHandle pos_dim = c->Dim(pos_shape, i); in __anonf38d6eaa0702()
274 "pos and len shapes must match: ", c->DebugString(pos_shape), in __anonf38d6eaa0702()
|