Searched refs:index_tensor (Results 1 – 3 of 3) sorted by relevance
167 const Tensor& index_tensor = context->input(0); in Compute() local168 OP_REQUIRES(context, TensorShapeUtils::IsScalar(index_tensor.shape()), in Compute()171 index_tensor.shape().DebugString())); in Compute()173 int32 index = index_tensor.scalar<int32>()(); in Compute()
348 Tensor index_tensor(DT_INT64, TensorShape({})); in InsertOneLocked() local
5483 auto index_tensor = in matchAndRewrite() local5486 op.getLoc(), index_tensor, shape_op); in matchAndRewrite()