Searched refs:serialized_shape (Results 1 – 4 of 4) sorted by relevance
222 U serialized_shape; in Compute() local223 OP_REQUIRES_OK(context, this->Serialize(output_shape, &serialized_shape)); in Compute()224 serialized_sparse_t.template chip<1>(2).setConstant(serialized_shape); in Compute()
228 const string& serialized_shape, DataType values_dtype, int index, in GetAndValidateSparseTensor() argument264 TF_RETURN_IF_ERROR(this->Deserialize(serialized_shape, output_shape)); in GetAndValidateSparseTensor()
277 const Variant& serialized_shape, in GetAndValidateSparseTensorShape() argument281 *output_shape = serialized_shape.get<Tensor>(); in GetAndValidateSparseTensorShape()
1106 serialized_shape = serialized.get_shape()1107 if serialized_shape.ndims is not None:1108 if serialized_shape.ndims != 0: