Searched refs:index_tensor (Results 1 – 2 of 2) sorted by relevance
181 const Tensor& index_tensor = context->input(0); in Compute() local182 OP_REQUIRES(context, TensorShapeUtils::IsScalar(index_tensor.shape()), in Compute()185 index_tensor.shape().DebugString())); in Compute()187 int32 index = index_tensor.scalar<int32>()(); in Compute()
348 Tensor index_tensor(DT_INT64, TensorShape({})); in InsertOneLocked() local