Searched refs:dense_shape_t (Results 1 – 2 of 2) sorted by relevance
45 const Tensor* dense_shape_t; in Compute() local49 OP_REQUIRES_OK(context, context->input("dense_shape", &dense_shape_t)); in Compute()54 OP_REQUIRES(context, TensorShapeUtils::IsVector(dense_shape_t->shape()), in Compute()56 dense_shape_t->shape().DebugString())); in Compute()73 const auto dense_shape = dense_shape_t->vec<int64>(); in Compute()
74 dense_shape_t = constant_op.constant(dense_shape, dtype=dtypes.int64)78 indices=indices_t, values=values_t, dense_shape=dense_shape_t)