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_t index = index_tensor.scalar<int32>()(); in Compute()
347 Tensor index_tensor(DT_INT64, TensorShape({})); in InsertOneLocked() local
6388 auto index_tensor = in matchAndRewrite() local6391 rewriter.create<shape::ShapeOfOp>(op.getLoc(), index_tensor, input); in matchAndRewrite()